Home
last modified time | relevance | path

Searched defs:all_tensors (Results 1 – 25 of 44) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/
Dnnapi_op.cc44 int NNAPIOp::InitNNAPIOpInOut(const std::vector<mindspore::MSTensor> &all_tensors) { in InitNNAPIOpInOut()
60 …TensorToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddTensorToNNAPIModel()
79 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dconv_nnapi.cc57 …dAttributesForConv(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddAttributesForConv()
114 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
154 std::vector<mindspore::MSTensor> *all_tensors) { in TransConvWeightFromKHWCToCHWK()
Dtranspose_nnapi.cc28 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
Dtopk_nnapi.cc33 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dstrided_slice_nnapi.cc36 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
Dreshape_nnapi.cc30 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dpadding_nnapi.cc40 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dgather_nnapi.cc36 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dconcat_nnapi.cc39 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dcast_nnapi.cc33 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dactivation_nnapi.cc49 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
Dreduce_nnapi.cc42 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dsoftmax_nnapi.cc41 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dinstance_norm_nnapi.cc46 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
Dfull_connection_nnapi.cc38 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
Dscale_nnapi.cc38 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Darithmetic_nnapi.cc67 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
Dsplit_nnapi.cc56 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dresize_nnapi.cc67 …:AddOpToNNAPIModel(ANeuralNetworksModel *nnapi_model, std::vector<mindspore::MSTensor> *all_tensor… in AddOpToNNAPIModel()
Dconv_transpose_nnapi.cc53 std::vector<mindspore::MSTensor> *all_tensors) { in AddOpToNNAPIModel()
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Darithmetic_grad_fp32_tests.cc198 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_1_dy… in TEST_F() local
241 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_1_dy… in TEST_F() local
287 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_8_dy… in TEST_F() local
334 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_2_dy… in TEST_F() local
381 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_3_dy… in TEST_F() local
426 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_4_dy… in TEST_F() local
480 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_4_dy… in TEST_F() local
526 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_9_dy… in TEST_F() local
572 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_9_dy… in TEST_F() local
618 …std::vector<lite::Tensor *> all_tensors = GenerateTensorsForTest("./operators/arithmetic_fp32_5_dy… in TEST_F() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/
Dsoftmax_npu.cc52 int SoftmaxNPUOp::HandleAxisAndConstantInputs(std::vector<mindspore::MSTensor *> *all_tensors) { in HandleAxisAndConstantInputs()
Dunpack_npu.cc56 int UnpackNPUOp::HandleAxisAndConstantInputs(std::vector<mindspore::MSTensor *> *all_tensors) { in HandleAxisAndConstantInputs()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Deliminate_transpose.cc119 …n(kernel::SubGraphKernel *subgraph, kernel::KernelExec *kernel, std::vector<Tensor *> *all_tensors, in DoPreFusion()
144 …el::SubGraphKernel *subgraph, const kernel::KernelExec *kernel, std::vector<Tensor *> *all_tensors, in DoPostFusion()
182 …e::EliminateForSingleKernel(kernel::SubGraphKernel *subgraph, std::vector<Tensor *> *all_tensors) { in EliminateForSingleKernel()
Dpass_utils.cc94 …e(kernel::SubGraphKernel *subgraph, kernel::KernelExec *kernel, std::vector<Tensor *> *all_tensors, in InsertPreTranspose()
130std::vector<Tensor *> *all_tensors, const TransInfoPair &trans_info, const size_t &index, in InsertPostTranspose()

12