utils.h

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
namespace detail {
std::vector<int> _computeStrides(std::vector<int> newShape);
 
std::vector<int> _broadcastShape(std::vector<int> a, std::vector<int> b);
 
std::vector<int> _broadcastStrides(std::vector<int> originalShape,
                                   std::vector<int> originalStrides,
                                   std::vector<int> targetShape);
 
int _computeOffset(std::vector<int> index, std::vector<int> strides);
} // namespace detail