| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | logit_cpu_kernel.h | 48 std::vector<int64_t> input_shape_; variable
|
| D | logit_grad_cpu_kernel.h | 48 std::vector<int64_t> input_shape_; variable
|
| D | triu_cpu_kernel.h | 44 std::vector<int64_t> input_shape_; variable
|
| D | tril_cpu_kernel.h | 48 std::vector<int64_t> input_shape_; variable
|
| D | layer_norm_grad_grad_cpu_kernel.h | 46 std::vector<int64_t> input_shape_; variable
|
| D | bias_add_grad_cpu_kernel.h | 50 std::vector<size_t> input_shape_; variable
|
| D | ps_roi_pooling_grad_cpu_kernel.h | 43 std::vector<int64_t> input_shape_; variable
|
| D | matrix_inverse_cpu_kernel.h | 52 ShapeVector input_shape_; variable
|
| D | fill_diagonal_cpu_kernel.h | 44 std::vector<int64_t> input_shape_; variable
|
| D | mirror_pad_grad_cpu_kernel.h | 104 std::vector<int64_t> input_shape_; variable
|
| D | slice_cpu_kernel.h | 51 std::vector<int64_t> input_shape_; variable
|
| D | reduce_std_cpu_kernel.h | 56 std::vector<int64_t> input_shape_; variable
|
| D | bias_add_cpu_kernel.h | 58 std::vector<size_t> input_shape_; variable
|
| D | self_adjoint_eig_cpu_kernel.h | 50 std::vector<int64_t> input_shape_; variable
|
| D | l2loss_cpu_kernel.h | 56 std::vector<int64_t> input_shape_; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | assign_sub_gpu_kernel.cc | 39 …std::vector<int64_t> input_shape_ = std::vector<int64_t>(inputs.at(kIndex0)->GetDeviceShapeVector(… in Init() local
60 …std::vector<int64_t> input_shape_ = std::vector<int64_t>(inputs.at(kIndex0)->GetDeviceShapeVector(… in Resize() local
|
| D | assign_add_gpu_kernel.cc | 40 …std::vector<int64_t> input_shape_ = std::vector<int64_t>(inputs.at(kIndex0)->GetDeviceShapeVector(… in Init() local
61 …std::vector<int64_t> input_shape_ = std::vector<int64_t>(inputs.at(kIndex0)->GetDeviceShapeVector(… in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | insert_transdata_for_runop.h | 43 ShapeVector input_shape_; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ps_roi_pooling_grad_v2_gpu_kernel.h | 44 std::vector<int64_t> input_shape_; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | expand_cpu_kernel.h | 55 std::vector<size_t> input_shape_; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | dropout3d.h | 31 std::vector<int64_t> input_shape_; variable
|
| D | dropout2d.h | 33 std::vector<int64_t> input_shape_; variable
|
| D | gather_d_grad_v2.h | 38 std::vector<int64_t> input_shape_; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/ |
| D | gather_grad_kernels.h | 36 std::vector<int64_t> input_shape_; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | softmax_grad_fp32_tests.cc | 41 int input_shape_[] = {1, 9, 11, 12}; in TEST_F() local
106 int input_shape_[] = {1, 9, 11, 12}; in TEST_F() local
176 int input_shape_[] = {1, 9, 11, 12}; in TEST_F() local
246 int input_shape_[] = {1, 9, 11, 12}; in TEST_F() local
315 int input_shape_[] = {1, 9, 11, 12}; in TEST_F() local
|