| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | clamp_tensor.cc | 80 auto max_shape = input2_shape->GetShapeVector(); in InferShape() local
111 auto max_shape = input2->shape(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/fasterrcnn/src/utils/ |
| D | box_utils.py | 32 def delta2bbox(deltas, boxes, weights=(10.0, 10.0, 5.0, 5.0), max_shape=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | pad_and_shift.cc | 55 ShapeVector max_shape = shape->GetShapeVector(); in InferShape() local
|
| D | all_swap.cc | 53 ShapeVector max_shape = {infer_max_size / tensor_in_shape[1], tensor_in_shape[1]}; in InferShape() local
|
| D | compute_accidental_hits.cc | 69 auto max_shape = std::make_shared<abstract::Shape>(max_shape_vec); in ComputeAccidentalHitsInferShape() local
|
| D | dynamic_broadcast_gradient_args.cc | 79 ShapeVector max_shape; in InferShape() local
|
| D | set_size.cc | 82 ShapeVector max_shape(dense_size[0] - 1); in SetSizeInferShape() local
|
| D | normalize_tuple_index.cc | 277 ShapeVector max_shape; in InferShape() local
|
| D | op_utils.cc | 68 const std::vector<int64_t> &max_shape = res ? x_shape : y_shape; in CalBroadCastShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ |
| D | dshape.h | 169 …TensorShape(const ShapeVector &list, const ShapeVector &max_shape) : shape_(list), max_shape_(max_… in TensorShape()
215 const ShapeVector &max_shape() const { return max_shape_; } in max_shape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | bounding_box_decode_tensorrt.h | 44 const std::vector<int> &max_shape, float wh_ratio_clip) in BoundingBoxDecodePlugin()
|
| D | bounding_box_decode_tensorrt.cc | 50 std::vector<int64_t> max_shape = GetValue<std::vector<int64_t>>(max_shape_attr); in AddInnerOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _quant_ops.py | 176 def infer_shape(self, x_shape, min_shape, max_shape): argument
237 def infer_shape(self, x_shape, min_shape, max_shape): argument
614 def infer_shape(self, x_shape, min_shape, max_shape): argument
678 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
730 def infer_shape(self, x_shape, min_shape, max_shape): argument
788 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
888 def infer_shape(self, x_shape, min_shape, max_shape): argument
934 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
1014 def infer_shape(self, x_shape, min_shape, max_shape): argument
1069 def infer_shape(self, dout_shape, x_shape, min_shape, max_shape): argument
|
| D | other_ops.py | 179 …def __init__(self, max_shape, means=(0.0, 0.0, 0.0, 0.0), stds=(1.0, 1.0, 1.0, 1.0), wh_ratio_clip… argument
|
| D | _inner_ops.py | 2528 def infer_shape(self, x_shape, pre_shape, min_shape, max_shape, n_step, loss_scale_shape): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/ |
| D | parameterized_truncated_normal_gpu_kernel.cc | 83 …std::vector<int64_t> max_shape = std::vector<int64_t>(inputs.at(kIndex4)->GetDeviceShapeVector().b… in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/ |
| D | boundingbox_decode_gpu_kernel.cc | 62 auto max_shape = primitive_->GetAttr("max_shape"); in Init() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_clamp.cc | 37 ShapeVector max_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | image_func.py | 29 def bounding_box_decode(anchor_box, deltas, max_shape, means=(0.0, 0.0, 0.0, 0.0), stds=(1.0, 1.0, … argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | matmul_avx512_mask_fp32.c | 106 int max_shape[C4NUM] = {C12NUM, C12NUM, C8NUM, C6NUM}; in MatMulMaskAvx512Fp32() local
|
| D | matmul_avx512_fp32.c | 85 int max_shape[C4NUM] = {C12NUM, C12NUM, C8NUM, C6NUM}; in MatMulAvx512Fp32() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runtime_info.cc | 37 auto max_shape = common::AnfAlgo::GetOutputMaxShape(node, output_index); in OpRuntimeInfoGetOutputTensorMemSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | boundingbox_decode_cpu_kernel.cc | 68 auto max_shape = primitive_->GetAttr("max_shape"); in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | session_basic.cc | 145 auto max_shape = common::AnfAlgo::GetOutputMaxShape(node, output_index); in CreateNodeOutputTensor() local
|
| D | anf_runtime_algorithm.cc | 355 auto max_shape = common::AnfAlgo::GetOutputMaxShape(node, output_index); in GetOutputTensorMemSize() local
|