Home
last modified time | relevance | path

Searched defs:input_shapes_ (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dconcat_offset.h33 std::vector<std::vector<int64_t>> input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/
Dquant_dtype_cast_kernel.h36 std::vector<int64_t> input_shapes_; variable
Dsequence_addn.h38 std::vector<std::vector<int64_t>> input_shapes_; variable
Dsequence_concat.h41 std::vector<std::vector<int64_t>> input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dprint_cpu_kernel.h52 std::vector<ShapeVector> input_shapes_; variable
Dconcat_offset_cpu_kernel.h53 std::vector<ShapeVector> input_shapes_; variable
Dgathernd_cpu_kernel.h56 ShapeVector input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dconcat_int8.h61 int **input_shapes_ = nullptr; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dmeshgrid_gpu_kernel.h67 std::vector<int64_t> input_shapes_; variable
Dgathernd_gpu_kernel.h105 std::vector<int64_t> input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/wrapper/int8/
Dconcat_int8_wrapper.h34 int **input_shapes_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Ddense_to_csr_sparse_matrix_gpu_kernel.h60 std::vector<size_t> input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dsparse_add_grad_helper.h174 std::vector<std::vector<int64_t>> input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Deinsum_gpu_kernel.h213 std::vector<std::vector<int64_t>> input_shapes_; variable
Deinsum_grad_gpu_kernel.h396 std::vector<std::vector<int64_t>> input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dconvert.h366 ShapeArray input_shapes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/akg/
Dakg_kernel_json_decoder.cc316 std::vector<ShapeVector> input_shapes_; member in mindspore::kernel::__anon39e4c91c0111::CNodeDecoder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dquantize.h55 int32_t **input_shapes_; member