| /third_party/mindspore/mindspore/core/ops/ |
| D | dynamic_broadcast_gradient_args.cc | 39 …auto max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_arg->BuildShape())[kMaxShap… in CheckInputsAndGetShape() local
66 ShapeVector max_shape{max_size}; in Infer() local
|
| D | log1p.cc | 34 auto max_shape = shape_map[kMaxShape]; in Log1pInferShape() local
|
| D | softplus.cc | 34 auto max_shape = shape_map[kMaxShape]; in SoftplusInferShape() local
|
| D | gelu.cc | 38 auto max_shape = shape_map[kMaxShape]; in InferShape() local
|
| D | exp.cc | 40 auto max_shape = shape_map[kMaxShape]; in InferShape() local
|
| D | reshape.cc | 73 ShapeVector max_shape = shape_v; in ReshapeInfer() local
|
| D | log.cc | 40 auto max_shape = shape_map[kMaxShape]; in InferShape() local
|
| D | neg.cc | 40 auto max_shape = shape_map[kMaxShape]; in InferShape() local
|
| D | reciprocal.cc | 40 auto max_shape = shape_map[kMaxShape]; in InferShape() local
|
| D | embedding_lookup.cc | 61 ShapeVector max_shape; in EmbeddingLookupInfer() local
|
| D | log_softmax.cc | 48 auto max_shape = shape_map[kMaxShape]; in LogSoftmaxInferShape() local
|
| D | fake_quant_with_min_max_vars_per_channel.cc | 49 …auto max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildSh… in FakeQuantWithMinMaxVarsPerChannelInfer() local
|
| D | addn.cc | 61 auto max_shape = element0_shape_map[kMaxShape]; in AddNInferShape() local
|
| D | softmax.cc | 60 auto max_shape = shape_map[kMaxShape]; in SoftMaxInferShape() local
|
| D | one_hot.cc | 37 auto max_shape = shape_map[kMaxShape]; in OneHotInferShape() local
|
| D | getnext.cc | 68 std::vector<ShapeVector> max_shape; in GetnextInferShape() local
|
| D | gather.cc | 92 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in GatherInfer() local
|
| D | fake_quant_with_min_max_vars.cc | 34 …auto max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildSh… in InferShape() local
|
| D | transpose.cc | 73 std::vector<int64_t> max_shape; in InferShape() local
|
| D | square.cc | 40 auto max_shape = shape_map[kMaxShape]; in SquareInferShape() local
|
| D | tile.cc | 63 auto max_shape = shape_map[kMaxShape]; in TileInferShape() local
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 148 ShapeVector max_shape = shape->max_shape(); in InferImplUnique() local
194 ShapeVector max_shape = shape->max_shape(); in InferImplPadAndShift() local
260 ShapeVector max_shape; in InferImplUnsortedSegmentSum() local
313 ShapeVector max_shape; in InferImplUnsortedSegmentMax() local
365 ShapeVector max_shape; in InferImplUnsortedSegmentMin() local
395 ShapeVector max_shape = x->shape()->max_shape(); in InferImplScatterAdd() local
410 ShapeVector max_shape = x->shape()->max_shape(); in InferImplScatterSub() local
424 ShapeVector max_shape = x->shape()->max_shape(); in InferImplScatterUpdate() local
443 ShapeVector max_shape; in InferImplMapCacheIdx() local
485 ShapeVector max_shape; in InferImplCacheSwapTable() local
[all …]
|
| D | dshape.h | 72 Shape(const ShapeVector &list, const ShapeVector &min_shape, const ShapeVector &max_shape) in Shape()
84 const ShapeVector &max_shape() { return max_shape_; } in max_shape() function
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | bias_add_grad.cc | 51 auto max_shape = shape_map[kMaxShape]; in InferShape() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _quant_ops.py | 105 def infer_shape(self, x_shape, min_shape, max_shape): argument
165 def infer_shape(self, x_shape, min_shape, max_shape): argument
541 def infer_shape(self, x_shape, min_shape, max_shape): argument
605 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
657 def infer_shape(self, x_shape, min_shape, max_shape): argument
715 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
815 def infer_shape(self, x_shape, min_shape, max_shape): argument
861 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
944 def infer_shape(self, x_shape, min_shape, max_shape): argument
999 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
|