38 const std::vector<Tensor>& index_tensors,
39 const SizeVector& indexed_shape,
40 const SizeVector& indexed_strides);
44 const std::vector<Tensor>& index_tensors,
45 const SizeVector& indexed_shape,
46 const SizeVector& indexed_strides);
48 #ifdef BUILD_CUDA_MODULE 49 void IndexGetCUDA(
const Tensor& src,
51 const std::vector<Tensor>& index_tensors,
52 const SizeVector& indexed_shape,
53 const SizeVector& indexed_strides);
58 const std::vector<Tensor>& index_tensors,
59 const SizeVector& indexed_shape,
60 const SizeVector& indexed_strides);
64 const std::vector<Tensor>& index_tensors,
65 const SizeVector& indexed_shape,
66 const SizeVector& indexed_strides);
68 #ifdef BUILD_CUDA_MODULE 69 void IndexSetCUDA(
const Tensor& src,
71 const std::vector<Tensor>& index_tensors,
72 const SizeVector& indexed_shape,
73 const SizeVector& indexed_strides);
void IndexSetCPU(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSetCPU.cpp:73
void IndexSet(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSet.cpp:66
void IndexGetCPU(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSetCPU.cpp:51
Definition: PinholeCameraIntrinsic.cpp:35
void IndexGet(const Tensor &src, Tensor &dst, const std::vector< Tensor > &index_tensors, const SizeVector &indexed_shape, const SizeVector &indexed_strides)
Definition: IndexGetSet.cpp:40