Searched refs:kMaxShape (Results 1 – 25 of 28) sorted by relevance
12
54 if (x_shape[kMaxShape] != cond_shape[kMaxShape] || x_shape[kMaxShape] != y_shape[kMaxShape]) { in SelectInferShape()
66 auto x_max_shape = x_shape_map[kMaxShape]; in BroadCastInferShape()68 auto y_max_shape = y_shape_map[kMaxShape]; in BroadCastInferShape()
59 ShapeVector x_max_shape = x_shape_map[kMaxShape]; in MatMulInferShape()61 ShapeVector y_max_shape = y_shape_map[kMaxShape]; in MatMulInferShape()
71 ShapeVector x_max_shape = x_shape_map[kMaxShape]; in BatchMatmulInferShape()73 ShapeVector y_max_shape = y_shape_map[kMaxShape]; in BatchMatmulInferShape()
34 auto max_shape = shape_map[kMaxShape]; in Log1pInferShape()
34 auto max_shape = shape_map[kMaxShape]; in SoftplusInferShape()
38 auto max_shape = shape_map[kMaxShape]; in InferShape()
40 auto max_shape = shape_map[kMaxShape]; in InferShape()
48 auto max_shape = shape_map[kMaxShape]; in LogSoftmaxInferShape()
39 …to max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_arg->BuildShape())[kMaxShape]; in CheckInputsAndGetShape()
60 auto max_shape = shape_map[kMaxShape]; in SoftMaxInferShape()
63 auto max_shape = shape_map[kMaxShape]; in TileInferShape()
40 auto max_shape = shape_map[kMaxShape]; in SquareInferShape()
37 auto max_shape = shape_map[kMaxShape]; in OneHotInferShape()
33 …ax_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kMaxShape]; in InferShape()
66 auto max_shape = shape_map[kMaxShape]; in InferShape()
61 auto max_shape = element0_shape_map[kMaxShape]; in AddNInferShape()
75 auto max_shape = input_shape[kMaxShape]; in InferShape()
43 auto max_shape = shape_map[kMaxShape]; in InferShape()
160 auto x_max_shape = x_shape_map[kMaxShape]; in Conv2dInferShape()162 auto w_max_shape = w_shape_map[kMaxShape]; in Conv2dInferShape()
51 auto max_shape = shape_map[kMaxShape]; in InferShape()
37 constexpr auto kMaxShape = "max_shape"; variable