Searched defs:x_rank (Results 1 – 19 of 19) sorted by relevance
35 size_t x_rank = value_list.size(); in CalDynamicOutputShape() local68 size_t x_rank = shape_list.size(); in CalOutputShape() local
36 auto x_rank = SizeToLong(x_shape.size()); in InferShape() local
50 auto x_rank = SizeToLong(x_shape.size()); in L2NormalizeInfer() local
47 auto x_rank = SizeToLong(x_shape.size()); in ArgMinInfer() local
29 const int64_t x_rank = 2; in InferShape() local
62 const int64_t x_rank = 4; in DepthToSpaceInfer() local
34 int64_t x_rank = SizeToLong(x_shape.size()); in GatherDInferShape() local
63 const int64_t x_rank = 4; in BatchToSpaceInfer() local
27 auto x_rank = SizeToLong(x_shape.size()); in InferShape() local
36 auto x_rank = SizeToLong(x_shape.size()); in IndexAddInferShape() local
33 const int64_t x_rank = 4; in InferShape() local
32 auto x_rank = SizeToLong(x_shape.size()); in InferShape() local
65 size_t x_rank = x_shape.size(); in EllipsisInferShape() local147 size_t x_rank = x_shape.size(); in ComputeInferShape() local
91 const int64_t x_rank = 4; in InferShape() local
78 size_t x_rank = x_shape.size(); in Init() local
45 def _cal_repeat_dims(x_rank, rep, expand_axis): argument
65 const int64_t x_rank = 4; in InferShape() local
217 size_t x_rank = x_shape->size(); in DoInferReduceShape() local