| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | splice_fp32_coder.cc | 29 std::vector<int> src_shape = input_tensor_->shape(); in DoCode() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | splice_fp32.cc | 44 std::vector<int> src_shape = input_tensor->shape(); in Run() local
|
| D | affine_fp32.cc | 125 std::vector<int> src_shape = in_tensors_.at(kInputIndex)->shape(); in FullRunInit() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | log_softmax_cpu_kernel.cc | 33 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | log_softmax_grad_cpu_kernel.cc | 33 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | softmax_cpu_kernel.cc | 33 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | eltwise_cpu_kernel.cc | 61 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | addn_cpu_kernel.cc | 102 auto src_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, index); in CheckParam() local
|
| D | pooling_avg_grad_cpu_kernel.cc | 38 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | pooling_cpu_kernel.cc | 38 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | conv2d_grad_filter_cpu_kernel.cc | 36 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1); in InitKernel() local
|
| D | conv2d_grad_input_cpu_kernel.cc | 38 std::vector<size_t> src_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | conv_cpu_kernel.cc | 37 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | lstm_cpu_kernel.cc | 117 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in CheckParam() local
|
| D | mkl_cpu_kernel.cc | 27 … const std::vector<size_t> &src_shape, const std::vector<size_t> &kernel_size, in GetPadding()
|
| D | lstm_grad_cpu_kernel.cc | 109 std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in CheckParam() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reshape_eliminate.h | 53 auto src_shape = src_shape_abs->GetShapeTrack(); in operator() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | index_add_gpu_kernel.h | 73 std::vector<size_t> src_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/ |
| D | npu_converter_utils.cc | 57 ge::Shape ConverterToNPUShape(const std::vector<int64_t> &src_shape) { in ConverterToNPUShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | bce_with_logits_loss_impl.cu | 30 …global__ void FillAndBroadcast(const size_t size, const size_t shape_size, const size_t *src_shape, in FillAndBroadcast()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | common_infer.c | 294 void ShapeSet(int *dst_shape, size_t *dst_shape_size, const int *src_shape, size_t src_shape_size) { in ShapeSet()
|