Searched defs:data_dims (Results 1 – 4 of 4) sorted by relevance
33 size_t NormalizeDimIndex::ConstNormalizeDimIndex(size_t data_dims, size_t dim_index, in ConstNormalizeDimIndex()
664 size_t data_dims = data_shape.size(); in TensorGetitemByTuple() local1394 …ex::SetItemByTensorByBool(const ShapeVector &data_shape, const TensorPtr &index, int64_t data_dims, in SetItemByTensorByBool()1479 size_t GetSpecifiedDimensions(const py::tuple &new_tuple_index, size_t data_dims) { in GetSpecifiedDimensions()1518 size_t data_dims = data_shape.size(); in GetItemByTupleWithView() local2015 int64_t data_dims = SizeToLong(data_shape.size()); in SetItemByTensor() local
571 static inline bool UseCopySlice(const std::vector<TensorIndex> &indices, int64_t data_dims) { in UseCopySlice()
338 static size_t NormalizeDimIndex(const size_t data_dims, size_t dim_index, in NormalizeDimIndex()