| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_broadcast_selecter.cc | 101 auto shape_tmp = input_shapes_[0]; in IsBroadCastSupport5HD() local
104 …[&shape_tmp](const std::vector<size_t> &elem) { return shape_tmp.at(kChannelC) != elem.at(kChannel… in IsBroadCastSupport5HD()
183 auto shape_tmp = input_shapes_[0]; in IsBroadCastSupportC1HWNCoC0() local
185 …std::any_of(input_shapes_.begin(), input_shapes_.end(), [&shape_tmp](const std::vector<size_t> &el… in IsBroadCastSupportC1HWNCoC0()
186 …return (shape_tmp.at(kChannelC) != elem.at(kChannelC) || shape_tmp.at(kChannelN) != elem.at(kChann… in IsBroadCastSupportC1HWNCoC0()
235 auto shape_tmp = input_shapes_[0]; in IsBroadCastSupportFracNZ() local
237 …std::any_of(input_shapes_.begin(), input_shapes_.end(), [&shape_tmp](const std::vector<size_t> &el… in IsBroadCastSupportFracNZ()
238 return (shape_tmp.at(shape_tmp.size() - 1) != elem.at(elem.size() - 1)) || in IsBroadCastSupportFracNZ()
239 (shape_tmp.at(shape_tmp.size() - 2) != elem.at(elem.size() - 2)); in IsBroadCastSupportFracNZ()
283 auto shape_tmp = input_shapes_[0]; in IsBroadCastSupportNDC1HWC0() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | concat_infer.c | 74 int shape_tmp[MAX_SHAPE_SIZE] = {0}; in ConcatInferShape() local
76 ShapeSet(shape_tmp, &shape_tmp_size, inputs[i]->shape_, inputs[i]->shape_size_); in ConcatInferShape()
81 int axis_tmp = shape_tmp[axis]; in ConcatInferShape()
82 erase_ret = ShapeErase(shape_tmp, &shape_tmp_size, axis); in ConcatInferShape()
86 …if (!ShapeEqual(input0_shape_without_axis, input0_shape_without_axis_size, shape_tmp, shape_tmp_si… in ConcatInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | dynamic_assign_cpu_kernel.cc | 99 ShapeVector shape_tmp; in LaunchKernel() local
100 …(void)std::transform(input_x_shape.begin(), input_x_shape.end(), std::back_inserter(shape_tmp), Si… in LaunchKernel()
101 tensor->set_shape(shape_tmp); in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_rnn_grad_fission_v2.cc | 331 std::vector<size_t> shape_tmp; in CreateHConcat() local
333 shape_tmp = origin_input4_shape; in CreateHConcat()
335 shape_tmp = {1, origin_input4_shape[0], origin_input4_shape[1]}; in CreateHConcat()
340 …InferTypeAndShape({AnfAlgo::GetOutputInferDataType(origin_input4, 0)}, {shape_tmp}, reshape.get()); in CreateHConcat()
387 std::vector<size_t> shape_tmp; in CreateConcatNodeT1() local
389 shape_tmp = origin_input4_shape; in CreateConcatNodeT1()
391 shape_tmp = {1, origin_input4_shape[0], origin_input4_shape[1]}; in CreateConcatNodeT1()
396 …InferTypeAndShape({AnfAlgo::GetOutputInferDataType(origin_input4, 0)}, {shape_tmp}, reshape.get()); in CreateConcatNodeT1()
404 origin_input0_shape[kDim2] + shape_tmp[kDim2]}; in CreateConcatNodeT1()
|
| D | dynamic_gru_v2_grad_fission.cc | 242 std::vector<std::vector<size_t>> shape_tmp; in CreateHReshape() local
244 shape_tmp = {ori_shape}; in CreateHReshape()
246 shape_tmp = {{IntToSize(1), ori_shape[kDim0], ori_shape[kDim1]}}; in CreateHReshape()
252 AnfAlgo::SetOutputInferTypeAndShape(ori_dtype, shape_tmp, reshape.get()); in CreateHReshape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.cc | 309 std::vector<size_t> shape_tmp; in UpdateGraphInputAbstract() local
310 …(void)std::transform(tensor_shape.begin(), tensor_shape.end(), std::back_inserter(shape_tmp), IntT… in UpdateGraphInputAbstract()
311 … AnfAlgo::SetOutputInferTypeAndShape({AnfAlgo::GetOutputInferDataType(input_node, 0)}, {shape_tmp}, in UpdateGraphInputAbstract()
313 size = abstract::ShapeSize(shape_tmp) * abstract::TypeIdSize(tensor->data_type()); in UpdateGraphInputAbstract()
|
| D | ascend_session.cc | 482 std::vector<size_t> shape_tmp; in LoadInputData() local
483 …(void)std::transform(tensor_shape.begin(), tensor_shape.end(), std::back_inserter(shape_tmp), IntT… in LoadInputData()
484 … AnfAlgo::SetOutputInferTypeAndShape({AnfAlgo::GetOutputInferDataType(input_node, 0)}, {shape_tmp}, in LoadInputData()
486 size = abstract::ShapeSize(shape_tmp) * abstract::TypeIdSize(tensor->data_type()); in LoadInputData()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 350 std::vector<size_t> shape_tmp; in BindInputTensorAddressPtr() local
351 …(void)std::transform(tensor_shape.begin(), tensor_shape.end(), std::back_inserter(shape_tmp), IntT… in BindInputTensorAddressPtr()
352 …AnfAlgo::SetOutputInferTypeAndShape({AnfAlgo::GetOutputInferDataType(item, 0)}, {shape_tmp}, item.… in BindInputTensorAddressPtr()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | tensor.cc | 653 std::vector<size_t> shape_tmp; in data_sync() local
654 (void)std::transform(shape().begin(), shape().end(), std::back_inserter(shape_tmp), IntToSize); in data_sync()
655 auto size = abstract::ShapeSize(shape_tmp) * abstract::TypeIdSize(data_type()); in data_sync()
|