| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 147 ShapeVector min_shape = {1}; in InferImplUnique() local
193 ShapeVector min_shape = {1}; in InferImplPadAndShift() local
259 ShapeVector min_shape; in InferImplUnsortedSegmentSum() local
312 ShapeVector min_shape; in InferImplUnsortedSegmentMax() local
364 ShapeVector min_shape; in InferImplUnsortedSegmentMin() local
394 ShapeVector min_shape = x->shape()->min_shape(); in InferImplScatterAdd() local
409 ShapeVector min_shape = x->shape()->min_shape(); in InferImplScatterSub() local
423 ShapeVector min_shape = x->shape()->min_shape(); in InferImplScatterUpdate() local
442 ShapeVector min_shape; in InferImplMapCacheIdx() local
486 ShapeVector min_shape; in InferImplCacheSwapTable() local
[all …]
|
| D | dshape.h | 72 Shape(const ShapeVector &list, const ShapeVector &min_shape, const ShapeVector &max_shape) in Shape()
83 const ShapeVector &min_shape() { return min_shape_; } in min_shape() function
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | log1p.cc | 33 auto min_shape = shape_map[kMinShape]; in Log1pInferShape() local
|
| D | softplus.cc | 33 auto min_shape = shape_map[kMinShape]; in SoftplusInferShape() local
|
| D | gelu.cc | 37 auto min_shape = shape_map[kMinShape]; in InferShape() local
|
| D | exp.cc | 39 auto min_shape = shape_map[kMinShape]; in InferShape() local
|
| D | reshape.cc | 74 ShapeVector min_shape = shape_v; in ReshapeInfer() local
|
| D | reciprocal.cc | 39 auto min_shape = shape_map[kMinShape]; in InferShape() local
|
| D | neg.cc | 39 auto min_shape = shape_map[kMinShape]; in InferShape() local
|
| D | log.cc | 39 auto min_shape = shape_map[kMinShape]; in InferShape() local
|
| D | embedding_lookup.cc | 69 ShapeVector min_shape; in EmbeddingLookupInfer() local
|
| D | log_softmax.cc | 47 auto min_shape = shape_map[kMinShape]; in LogSoftmaxInferShape() local
|
| D | dynamic_broadcast_gradient_args.cc | 64 ShapeVector min_shape{1L}; in Infer() local
|
| D | fake_quant_with_min_max_vars_per_channel.cc | 48 …auto min_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildSh… in FakeQuantWithMinMaxVarsPerChannelInfer() local
|
| D | addn.cc | 60 auto min_shape = element0_shape_map[kMinShape]; in AddNInferShape() local
|
| D | gather.cc | 91 ShapeVector min_shape = calc_shape(indices_shp_min, param_shp_min); in GatherInfer() local
|
| D | getnext.cc | 67 std::vector<ShapeVector> min_shape; in GetnextInferShape() local
|
| D | softmax.cc | 59 auto min_shape = shape_map[kMinShape]; in SoftMaxInferShape() local
|
| D | one_hot.cc | 36 auto min_shape = shape_map[kMinShape]; in OneHotInferShape() local
|
| D | fake_quant_with_min_max_vars.cc | 33 …auto min_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildSh… in InferShape() local
|
| D | transpose.cc | 72 std::vector<int64_t> min_shape; in InferShape() local
|
| D | square.cc | 39 auto min_shape = shape_map[kMinShape]; in SquareInferShape() local
|
| D | reluv2.cc | 65 auto min_shape = shape_map[kMinShape]; 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
534 def check_broadcast(self, min_shape, input_shape): argument
541 def infer_shape(self, x_shape, min_shape, max_shape): argument
598 def check_broadcast(self, min_shape, input_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
[all …]
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | bias_add_grad.cc | 50 auto min_shape = shape_map[kMinShape]; in InferShape() local
|