| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | transposed_update_fusion.cc | 41 auto perm_tensor = std::make_shared<tensor::Tensor>(kNumberTypeInt32, perm_shape); in CreatePermTensor() local
56 tensor::TensorPtr perm_tensor = CreatePermTensor(transposed); in CreatePermValueNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/ |
| D | transpose_coreml.cc | 21 auto perm_tensor = in_tensors_.at(1); in IsSupport() local
|
| D | transpose_coreml.h | 29 auto perm_tensor = in_tensors.at(1); in TransposeCoreMLOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/internal/ |
| D | transpose.cc | 31 auto perm_tensor = inputs.at(kIndex1); // input 1 : perm in CreateOpParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | transpose_fp32_tests.cc | 37 lite::Tensor perm_tensor(kNumberTypeInt32, {10}); variable
80 lite::Tensor perm_tensor(kNumberTypeInt32, {10}); variable
238 lite::Tensor perm_tensor(kNumberTypeInt32, {5}); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/ |
| D | transpose_fp16_coder.cc | 61 auto perm_tensor = input_tensors_.at(SECOND_INPUT); in ResetStatus() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/ |
| D | transpose_int8_coder.cc | 35 auto perm_tensor = input_tensors_.at(1); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | transpose_dynamic_fp32_coder.cc | 125 auto perm_tensor = input_tensors_.at(SECOND_INPUT); in ResetStatus() local
|
| D | transpose_fp32_coder.cc | 51 auto perm_tensor = input_tensors_.at(1); in ResetStatus() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/ |
| D | transpose_optimizer.cc | 77 tensor::TensorPtr perm_tensor = nullptr; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | transpose_infer.c | 66 const TensorC *perm_tensor = inputs[1]; in TransposeInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/ |
| D | format_recognition.cc | 59 auto perm_tensor = perm_para->default_param()->cast<tensor::TensorPtr>(); in GetTransposeFormat() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | transpose_base.cc | 106 auto perm_tensor = in_tensors_.at(SECOND_INPUT); in ResetStatus() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | transpose.cc | 103 auto perm_tensor = ctx.Input(kIndex1); in TransposeParamCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | graph_builder.cc | 48 auto perm_tensor = Tensor(perm); in Transpose() local
|
| D | op_node.cc | 494 auto perm_tensor = perm_node->As<inner::ConstTensorNode>()->data(); in InferFormat() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/ |
| D | pass_utils.cc | 173 auto perm_tensor = kernel->in_tensors().at(1); in GetTransposeInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | transpose.c | 84 TensorC *perm_tensor = transpose->base_.in_[SECOND_INPUT]; in ResetTransposeStatus() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/ |
| D | transform_op_optimizer.cc | 317 auto perm_tensor = std::make_shared<tensor::Tensor>(perm, kInt64); in GenTransformOp() local
329 auto perm_tensor = iter->second; in SetInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/ |
| D | graph_util.cc | 577 auto &perm_tensor = graph->allTensors.at(cnode->inputIndex.at(1)); in GetTransposePerm() local
|