| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_broadcast_to_ops.py | 57 ms_shape = (-1, -1, 5, 6)
60 output = P.BroadcastTo(ms_shape)(Tensor(x_np))
65 output = P.BroadcastTo(ms_shape)(Tensor(x1_np))
69 ms_shape = (2, 3, -1, -1)
72 output = P.BroadcastTo(ms_shape)(Tensor(x1_np))
86 ms_shape = (2, -1, 4, 5)
89 P.BroadcastTo(ms_shape)(Tensor(x_np))
92 ms_shape = (-1, 1, -1, -1)
95 P.BroadcastTo(ms_shape)(Tensor(x_np))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
| D | trt_utils.h | 65 …static nvinfer1::Dims MsDimsToTrtDims(const std::vector<size_t> &ms_shape, bool ignore_batch_dim =…
68 for (size_t i = offset; i < ms_shape.size(); ++i) {
69 trt_dims.d[i - offset] = SizeToInt(ms_shape[i]);
71 trt_dims.nbDims = ms_shape.size() - offset;
75 …static nvinfer1::Dims MsDimsToTrtDims(const ShapeVector &ms_shape, bool ignore_batch_dim = false) {
78 for (size_t i = offset; i < ms_shape.size(); ++i) {
79 trt_dims.d[i - offset] = LongToInt(ms_shape[i]);
81 trt_dims.nbDims = ms_shape.size() - offset;
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_broadcast_to_op.py | 87 ms_shape = (-1, 4, 5, 6)
90 output = P.BroadcastTo(ms_shape)(Tensor(x_np))
95 output = P.BroadcastTo(ms_shape)(Tensor(x1_np))
99 ms_shape = (2, 3, -1, 5)
102 output = P.BroadcastTo(ms_shape)(Tensor(x1_np))
116 ms_shape = (2, -1, 4, 5)
119 P.BroadcastTo(ms_shape)(Tensor(x_np))
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 85 auto ms_shape = AnfAlgo::GetOutputDeviceShape(input_node, input_index); in ConvertInputShapeAndType() local
93 ge::Format ge_format = ascend::GeTypesConvert::GetGeFormat(ms_format, ms_shape.size()); in ConvertInputShapeAndType()
94 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in ConvertInputShapeAndType()
115 auto ms_shape = AnfAlgo::GetOutputDeviceShape(node, i); in ConvertOutputShapeAndType() local
123 ge::Format ge_format = ascend::GeTypesConvert::GetGeFormat(ms_format, ms_shape.size()); in ConvertOutputShapeAndType()
124 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in ConvertOutputShapeAndType()
158 auto ms_shape = AnfAlgo::GetInputDeviceShape(node, index); in NewConstantOp() local
162 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in NewConstantOp()
167 ge::Format ge_format = ascend::GeTypesConvert::GetGeFormat(ms_format, ms_shape.size()); in NewConstantOp()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | all_to_all_v_calc_param.cc | 55 auto ms_shape = AnfAlgo::GetInputDeviceShape(cnode, i); in CalcOpParam() local
57 …size_t origin_mem_size = std::accumulate(ms_shape.begin(), ms_shape.end(), type_size, std::multipl… in CalcOpParam()
63 auto ms_shape = AnfAlgo::GetOutputDeviceShape(cnode, i); in CalcOpParam() local
65 …size_t origin_mem_size = std::accumulate(ms_shape.begin(), ms_shape.end(), type_size, std::multipl… in CalcOpParam()
|
| D | converter.cc | 177 auto ms_shape = AnfAlgo::GetInputDeviceShape(cnode, i); in GenerateStubGeNode() local
178 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in GenerateStubGeNode()
186 auto ms_shape = AnfAlgo::GetOutputDeviceShape(cnode, i); in GenerateStubGeNode() local
187 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in GenerateStubGeNode()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | sparse_apply_r_m_s_prop.cc | 41 auto ms_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(ms_shape_ptr)[kShape]; in InferShape() local
52 (void)same_shape_args_map.insert(std::make_pair("shape of ms ", ms_shape)); in InferShape()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 395 void SetDumpShape(const std::vector<size_t> &ms_shape, NotNull<aicpu::dump::Shape *> dump_shape) { in SetDumpShape() argument
396 for (auto &dim : ms_shape) { in SetDumpShape()
|