Searched defs:data_dim (Results 1 – 8 of 8) sorted by relevance
59 size_t data_dim = data_shape.size(); in NormalizeSliceInfo() local
473 const int64_t data_dim = SizeToLong(data_shape.size()); in GenerateNonZeroIndex() local536 const size_t data_dim = data_shape.size(); in GetStrideInfoFromTuple() local1417 const int64_t data_dim = SizeToLong(data_shape.size()); in GetItemByTensor() local1446 int64_t data_dim = SizeToLong(data_shape.size()); in GetItemByList() local1464 static void JudgeTupleIndexDim(int64_t data_dim, const std::vector<TensorIndex> &new_tuple_indexes)… in JudgeTupleIndexDim()1636 int64_t data_dim = SizeToLong(new_data_shape.size()); in GetItemByTuple() local1660 int64_t data_dim = SizeToLong(data_shape.size()); in GetItemByBool() local1676 int64_t data_dim = SizeToLong(data_shape.size()); in GetItemByNumber() local1723 int64_t data_dim = SizeToLong(data_shape.size()); in GetItemByNumberWithView() local1752 size_t data_dim = data_shape.size(); in GetItemBySlice() local
392 static inline void JudgeDataDim(int64_t data_dim, int64_t min_data_dim, int64_t max_data_dim) { in JudgeDataDim()
146 def _get_reshape_src_dim(data_dim, cmp_dim): argument
648 def judge_tuple_index_dim_check_error(index_dim, data_dim, x): argument
222 def judge_data_dim(data_dim, min_data_dim=0, max_data_dim=8): argument
694 def judge_tuple_index_dim_check_error(index_dim, data_dim): argument
788 auto data_dim = data_shape_.size(); in GetItemByTuple() local1178 size_t data_dim = data_shape_.size() + expand_dims_cnt; in ExpandDimsByTupleIndex() local