Searched defs:x_max_shape (Results 1 – 9 of 9) sorted by relevance
61 ShapeVector x_max_shape = x->shape()->max_shape(); in ReshapeInfer() local
66 auto x_max_shape = x_shape_map[kMaxShape]; in BroadCastInferShape() local
33 …auto x_max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kM… in InferShape() local
59 ShapeVector x_max_shape = x_shape_map[kMaxShape]; in MatMulInferShape() local
71 ShapeVector x_max_shape = x_shape_map[kMaxShape]; in BatchMatmulInferShape() local
160 auto x_max_shape = x_shape_map[kMaxShape]; in Conv2dInferShape() local
138 ShapeVector x_max_shape = input_x->shape()->max_shape(); in InferImplBatchNorm() local253 ShapeVector x_max_shape = input_x->shape()->max_shape(); in InferImplConv2D() local358 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplBiasAdd() local
288 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplMatMul() local355 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplBatchMatMul() local
758 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplReshape() local1095 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplArgMaxWithValue() local