Home
last modified time | relevance | path

Searched defs:input_shape_ (Results 1 – 25 of 51) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsplit_cpu_kernel.h52 std::vector<int> input_shape_; variable
Dspacetodepth_cpu_kernel.h38 std::vector<size_t> input_shape_; variable
Dbias_add_cpu_kernel.h39 std::vector<size_t> input_shape_; variable
Dmirror_pad_cpu_kernel.h49 std::vector<int64_t> input_shape_; variable
Dmirror_pad_grad_cpu_kernel.h64 std::vector<int64_t> input_shape_; variable
Dpad_cpu_kernel.h47 std::vector<size_t> input_shape_; variable
Ddropout_cpu_kernel.h42 std::vector<size_t> input_shape_; variable
Dtensor_copy_slices_cpu_kernel.h41 std::vector<int64_t> input_shape_; variable
Dgather_cpu_kernel.h41 std::vector<size_t> input_shape_; variable
Dstrided_slice_grad_cpu_kernel.h43 std::vector<int> input_shape_; variable
Dtranspose_cpu_kernel.h48 std::vector<size_t> input_shape_; variable
Dstridedslice_cpu_kernel.h56 std::vector<size_t> input_shape_; variable
Dslice_grad_cpu_kernel.h54 std::vector<size_t> input_shape_; variable
Dreduce_cpu_kernel.h42 std::vector<size_t> input_shape_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dtensor_copy_slices.h43 std::vector<int64_t> input_shape_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dsoftmax_parameter.h28 int input_shape_[DIMENSION_5D]; member
Dbroadcast_to_parameter.h28 int input_shape_[MAX_SHAPE_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dlayer_norm_info.h65 Shape input_shape_; variable
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dresize_int8_wrapper.h29 const int *input_shape_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dsoftmax_grad.h33 int input_shape_[5]; member
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_options.h47 std::string input_shape_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dspace_to_batch_fp32.h29 int input_shape_[4]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/
Dgpu_convert_to_dynamic_shape_gpu_kernel.h104 std::vector<size_t> input_shape_; variable
/third_party/mindspore/mindspore/core/ops/grad/
Dbias_add_grad.cc52 auto input_shape_ = GetFormatShape(format, input_shape); in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dreverse_v2_gpu_kernel.h147 std::vector<size_t> input_shape_; variable

123