Searched refs:kMinShape (Results 1 – 25 of 27) sorted by relevance
12
57 if (x_shape[kMinShape] != cond_shape[kMinShape] || x_shape[kMinShape] != y_shape[kMinShape]) { in SelectInferShape()
65 auto x_min_shape = x_shape_map[kMinShape]; in BroadCastInferShape()67 auto y_min_shape = y_shape_map[kMinShape]; in BroadCastInferShape()
58 ShapeVector x_min_shape = x_shape_map[kMinShape]; in MatMulInferShape()60 ShapeVector y_min_shape = y_shape_map[kMinShape]; in MatMulInferShape()
70 ShapeVector x_min_shape = x_shape_map[kMinShape]; in BatchMatmulInferShape()72 ShapeVector y_min_shape = y_shape_map[kMinShape]; in BatchMatmulInferShape()
33 auto min_shape = shape_map[kMinShape]; in Log1pInferShape()
33 auto min_shape = shape_map[kMinShape]; in SoftplusInferShape()
37 auto min_shape = shape_map[kMinShape]; in InferShape()
39 auto min_shape = shape_map[kMinShape]; in InferShape()
47 auto min_shape = shape_map[kMinShape]; in LogSoftmaxInferShape()
59 auto min_shape = shape_map[kMinShape]; in SoftMaxInferShape()
62 auto min_shape = shape_map[kMinShape]; in TileInferShape()
36 auto min_shape = shape_map[kMinShape]; in OneHotInferShape()
39 auto min_shape = shape_map[kMinShape]; in SquareInferShape()
32 …in_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kMinShape]; in InferShape()
65 auto min_shape = shape_map[kMinShape]; in InferShape()
60 auto min_shape = element0_shape_map[kMinShape]; in AddNInferShape()
74 auto min_shape = input_shape[kMinShape]; in InferShape()
42 auto min_shape = shape_map[kMinShape]; in InferShape()
159 auto x_min_shape = x_shape_map[kMinShape]; in Conv2dInferShape()161 auto w_min_shape = w_shape_map[kMinShape]; in Conv2dInferShape()
214 …in_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kMinShape]; in StridedSliceInferShape()
50 auto min_shape = shape_map[kMinShape]; in InferShape()
36 constexpr auto kMinShape = "min_shape"; variable