Home
last modified time | relevance | path

Searched defs:input_perm (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/view/
Dtest_broad_cast_to.cc34 auto input_perm = MakeValue(perm_data); in TEST_F() local
58 auto input_perm = MakeValue(input_perm_data); in TEST_F() local
Dtest_transpose.cc36 auto input_perm = MakeValue(input_perm_data); in TEST_F() local
Dtest_expand_dims.cc36 auto input_perm = MakeValue(input_perm_data); in TEST_F() local
Dtest_reshape.cc36 auto input_perm = MakeValue(input_perm_data); in TEST_F() local
Dtest_split.cc38 auto input_perm = MakeValue(perm_1); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dtranspose_gpu_kernel.cc76 …oseGpuKernelMod::GetPermValue(const std::vector<int64_t> &perm, std::vector<int32_t> *input_perm) { in GetPermValue()
117 std::vector<int32_t> input_perm; in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/view/
Dtranspose_strides_calc.cc27 const std::vector<int64_t> &input_perm) { in StridesCalc()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dtranspose_fp32_coder.cc126 auto input_perm = input_tensors_.at(1); in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dtranspose_impl.cuh57 … SimplifyTranspose(const std::vector<int64_t> &input_shape, const std::vector<int32_t> &input_perm, in SimplifyTranspose()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dgrad_math_ops.cc2110 auto input_perm = Range(SizeToLong(x_len)); in __anonc39942a70e02() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc1833 auto input_perm = node->input(kTwoNum); in ExportPrimTranspose() local