| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | broadcast_to_infer.c | 20 int GetShapeByType(const TensorC *shape_tensor, int shape_size, int *dst_shape) { in GetShapeByType() argument
21 if (shape_tensor == NULL || dst_shape == NULL) { in GetShapeByType()
32 dst_shape[i] = data[i]; in GetShapeByType()
38 dst_shape[i] = data[i]; in GetShapeByType()
44 dst_shape[i] = data[i]; in GetShapeByType()
50 dst_shape[i] = data[i]; in GetShapeByType()
56 dst_shape[i] = data[i]; in GetShapeByType()
154 int dst_shape[MAX_SHAPE_SIZE] = {0}; in BroadcastToInferShape() local
168 dst_shape[i] = param->shape_[i]; in BroadcastToInferShape()
176 ret = GetShapeByType(shape_tensor, dst_shape_size, dst_shape); in BroadcastToInferShape()
[all …]
|
| D | fill_infer.c | 35 const int32_t *dst_shape = (int32_t *)(dst_shape_tensor->data_); in FillInferShape() local
43 if (num_dims != 0 && dst_shape == NULL) { in FillInferShape()
52 ShapePush(output_shape, &output_shape_size, dst_shape[i]); in FillInferShape()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | broadcast_to_populate_v0.cc | 40 auto dst_shape = broadcast_to_prim->dst_shape(); in PopulateBroadcastToParameter() local
41 if (dst_shape == nullptr) { in PopulateBroadcastToParameter()
46 if (dst_shape->size() > MAX_SHAPE_SIZE) { in PopulateBroadcastToParameter()
51 broadcast_param->shape_size_ = dst_shape->size(); in PopulateBroadcastToParameter()
53 broadcast_param->shape_[i] = *(dst_shape->begin() + i); in PopulateBroadcastToParameter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | mkl_cpu_kernel.cc | 72 std::vector<size_t> *dst_shape) const { in BinaryBroadCast()
75 MS_EXCEPTION_IF_NULL(dst_shape); in BinaryBroadCast()
77 if (dst_shape->size() == 0) { in BinaryBroadCast()
78 (void)dst_shape->emplace_back(1); in BinaryBroadCast()
82 …"Binary broadcast in: src0: " << *src0_shape << " src1: " << *src1_shape << " dst: " << *dst_shape; in BinaryBroadCast()
83 if (src0_shape->size() != dst_shape->size()) { in BinaryBroadCast()
85 for (size_t i = src0_shape->size(); i < dst_shape->size(); ++i) { in BinaryBroadCast()
88 } else if (src1_shape->size() != dst_shape->size()) { in BinaryBroadCast()
89 for (size_t i = src1_shape->size(); i < dst_shape->size(); ++i) { in BinaryBroadCast()
111 << " dst: " << *dst_shape; in BinaryBroadCast()
[all …]
|
| D | addn_cpu_kernel.cc | 50 std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel() local
53 dnnl::memory::desc dst_mem_desc = GetDefaultMemDesc(dst_shape); in InitKernel()
97 auto dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in CheckParam() local
98 if (src0_shape != dst_shape) { in CheckParam()
|
| D | pooling_avg_grad_cpu_kernel.cc | 39 std::vector<size_t> dst_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1); in InitKernel() local
41 dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); in InitKernel()
|
| D | pooling_cpu_kernel.cc | 39 std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel() local
41 dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); in InitKernel()
|
| D | conv2d_grad_filter_cpu_kernel.cc | 38 std::vector<size_t> dst_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
53 dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); in InitKernel()
|
| D | conv_cpu_kernel.cc | 39 std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel() local
59 dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); in InitKernel()
|
| D | conv2d_grad_input_cpu_kernel.cc | 40 std::vector<size_t> dst_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
55 dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); in InitKernel()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | splice_fp32_coder.cc | 30 std::vector<int> dst_shape = output_tensor_->shape(); in DoCode() local
31 …if (src_shape.size() != dst_shape.size() || src_shape.size() != kInputSize2 || dst_shape.size() !=… in DoCode()
36 int dst_row = dst_shape.at(kWeightIndex); in DoCode()
38 int dst_col = dst_shape.at(kBiasIndex); in DoCode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_expand_parser.cc | 28 std::vector<int64_t> dst_shape; in Parse() local
34 dst_shape.push_back(static_cast<int64_t>(onnx_node_attr.ints(i))); in Parse()
51 dst_shape.emplace_back(data_ptr[i]); in Parse()
57 if (!dst_shape.empty()) { in Parse()
58 prim->set_shape(dst_shape); in Parse()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | splice_fp32.cc | 45 std::vector<int> dst_shape = output_tensor->shape(); in Run() local
46 …if (src_shape.size() != dst_shape.size() || src_shape.size() != DIMENSION_3D || dst_shape.size() !… in Run()
51 int dst_row = dst_shape.at(DIMENSION_1D); in Run()
53 int dst_col = dst_shape.at(DIMENSION_2D); in Run()
|
| D | affine_fp32.cc | 126 std::vector<int> dst_shape = full_input_->shape(); in FullRunInit() local
127 …if (src_shape.size() != dst_shape.size() || src_shape.size() != kInputSize2 || dst_shape.size() !=… in FullRunInit()
134 splice_dst_row_ = dst_shape.at(kInputRow); in FullRunInit()
135 splice_dst_col_ = dst_shape.at(kInputCol); in FullRunInit()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | broadcast_to_populate.cc | 39 auto dst_shape = value->shape(); in PopulateBroadcastToParameter() local
40 if (dst_shape == nullptr) { in PopulateBroadcastToParameter()
43 param->shape_size_ = dst_shape->size(); in PopulateBroadcastToParameter()
50 param->shape_[i] = dst_shape->Get(i); in PopulateBroadcastToParameter()
|
| /third_party/mindspore/mindspore/lite/src/ops/compat/v0/ |
| D | broadcast_to_compat_v0.cc | 40 auto dst_shape_attr = param->dst_shape(); in TransferBroadcastToAttr()
45 std::vector<int> dst_shape = std::vector<int>(dst_shape_attr->begin(), dst_shape_attr->end()); in TransferBroadcastToAttr() local
46 …auto dst_shape_tensor = AttrToTensor(dst_shape.data(), dst_shape.size(), true, kNumberTypeInt32, t… in TransferBroadcastToAttr()
|
| D | permute_compat_v0.cc | 45 std::vector<int> dst_shape; in TransferPermuteAttr() local
47 dst_shape.push_back(static_cast<int>(*it)); in TransferPermuteAttr()
49 …auto dst_shape_tensor = AttrToTensor(dst_shape.data(), dst_shape.size(), true, kNumberTypeInt32, t… in TransferPermuteAttr()
|
| D | nchw2nhwc_compat_v0.cc | 33 std::vector<int> dst_shape{0, 2, 3, 1}; // nchw to nhwc in TransferNchw2NhwcAttr() local
34 …auto dst_shape_tensor = AttrToTensor(dst_shape.data(), dst_shape.size(), true, kNumberTypeInt32, t… in TransferNchw2NhwcAttr()
|
| D | nhwc2nchw_compat_v0.cc | 33 std::vector<int> dst_shape{0, 3, 1, 2}; // nhwc to nchw in TransferNhwc2NchwAttr() local
34 …auto dst_shape_tensor = AttrToTensor(dst_shape.data(), dst_shape.size(), true, kNumberTypeInt32, t… in TransferNhwc2NchwAttr()
|
| D | transpose_compat_v0.cc | 45 std::vector<int> dst_shape = std::vector<int>(perm_attr->begin(), perm_attr->end()); in TransferTransposeAttr() local
46 …auto dst_shape_tensor = AttrToTensor(dst_shape.data(), dst_shape.size(), true, kNumberTypeInt32, t… in TransferTransposeAttr()
|
| D | reshape_compat_v0.cc | 46 std::vector<int> dst_shape = std::vector<int>(dst_shape_attr->begin(), dst_shape_attr->end()); in TransferReshapeAttr() local
47 …auto dst_shape_tensor = AttrToTensor(dst_shape.data(), dst_shape.size(), true, kNumberTypeInt32, t… in TransferReshapeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | index_add_gpu_kernel.h | 71 std::vector<size_t> dst_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
74 …is_null_input_ = CHECK_NULL_INPUT(dst_shape) || CHECK_NULL_INPUT(index_shape) || CHECK_NULL_INPUT(… in Init()
93 if (axis < 0 || axis >= SizeToInt(src_shape.size()) || axis >= SizeToInt(dst_shape.size())) { in Init()
98 dst_axis_size_ = dst_shape[axis]; in Init()
101 for (auto x : dst_shape) { in Init()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_broadcast_to_parser.cc | 36 std::vector<int64_t> dst_shape; in Parse() local
37 …TfliteData(tflite_op->inputs.at(1), tflite_subgraph->tensors, tflite_model->buffers, &dst_shape)) { in Parse()
41 prim->set_shape(dst_shape); in Parse()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_transpose_for_basiclstm_op.cc | 48 auto dst_shape = {origin_shape[1], origin_shape[0]}; in Insert() local
52 AnfAlgo::SetOutputInferTypeAndShape({origin_type}, {dst_shape}, transpose.get()); in Insert()
68 auto dst_shape = {origin_shape[0], origin_shape[1]}; in Insert() local
72 AnfAlgo::SetOutputInferTypeAndShape({dtype}, {dst_shape}, transpose.get()); in Insert()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | bce_with_logits_loss_impl.cu | 31 const size_t *dst_shape, const T *src, T *dst) { in FillAndBroadcast() argument
35 size_t pos_size = size / dst_shape[0]; in FillAndBroadcast()
39 pos_size = pos_size / dst_shape[i]; in FillAndBroadcast()
|