Searched defs:input_axis (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | conjugate_transpose_impl.cu | 28 … const size_t *output_stride, const S *input_axis, const size_t shape_size, in ConjugateTranspose()
53 … const size_t *output_stride, const S *input_axis, const size_t shape_size, in ConjugateTransposeComplex()
78 … const size_t *output_stride, const S *input_axis, const size_t shape_size, T *output, in CalConjugateTranspose()
87 … const size_t *output_stride, const S *input_axis, const size_t shape_size, in CalConjugateTransposeComplex()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | conjugate_transpose_gpu_kernel.cc | 134 S *input_axis = GetDeviceAddress<S>(inputs, kIndex1); in LaunchKernel() local
166 S *input_axis = GetDeviceAddress<S>(inputs, kIndex1); in LaunchComplexKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | unstack_ext.cc | 38 auto input_axis = MakeValue(axis); in UnstackExtAscendCustomize() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/view/ |
| D | test_squeeze.cc | 36 auto input_axis = MakeValue(axis_data); in TEST_F() local
|
| D | test_unstack.cc | 36 auto input_axis = MakeValue(axis_data); in TEST_F() local
|
| D | test_split.cc | 39 auto input_axis = MakeValue(axis_1); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | reversev2.cc | 85 auto input_axis = reinterpret_cast<int64_t *>(ctx.Input(1)->GetData()); in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | reduce_axis_update.cc | 180 const AnfNodePtr &input_axis = inputs.at(kAxisInputIndex); in CheckMatchedDAG() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1721 auto input_axis = node->input(kTwoNum); in ExportPrimReduce() local
1764 auto input_axis = node->input(kTwoNum); in ExportPrimReduceAnyOrAll() local
|