Home
last modified time | relevance | path

Searched defs:output_tensors (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/
Dop_coder.cc32 const std::vector<Tensor *> OperatorCoder::output_tensors() const { return output_tensors_; } in output_tensors() function in mindspore::lite::micro::OperatorCoder
/third_party/mindspore/mindspore/lite/test/ut/src/
Dutils_test.cc66 auto output_tensors = kernel::LiteKernelUtil::SubgraphOutputTensors(kernels); in TEST_F() local
/third_party/mindspore/mindspore/lite/src/delegate/nnrt/
Dnnrt_model_kernel.cc164 auto output_tensors = this->outputs(); in TransferOutputs() local
Dnnrt_delegate.cc276 auto output_tensors = model->outputs(); in PrepareOutputs() local
/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dinfershape_pass.cc49 void FreeTensors(std::vector<Tensor *> *input_tensors, std::vector<Tensor *> *output_tensors) { in FreeTensors()
249 …ensorShape(const std::vector<Tensor *> &input_tensors, const std::vector<Tensor *> &output_tensors in PrintTensorShape()
269 int SetDataType(MetaGraphT *graph, const std::vector<Tensor *> &output_tensors, std::vector<InferTe… in SetDataType()
512 auto output_tensors = ConvertTensorToLiteTensor(graph, node->outputIndex); in InferSubgraph() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconstant_folding_fusion.cc219 std::vector<Tensor *> output_tensors) { in ReplaceCNode()
356 std::vector<Tensor *> output_tensors; in Process() local
/third_party/mindspore/mindspore/lite/micro/coder/
Dgraph.cc193 std::vector<lite::Tensor *> CoderGraph::output_tensors() const { return output_tensors_; } in output_tensors() function in mindspore::lite::micro::CoderGraph
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel_util.cc120 std::vector<lite::Tensor *> output_tensors; in SubgraphOutputTensors() local
Dscheduler.cc1117 std::vector<Tensor *> output_tensors; in CreateSubGraphKernel() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc637 std::vector<tensor::TensorPtr> output_tensors; in RunControlOperator() local
857 auto &output_tensors = actor_set->output_actor_->outputs(); in RunGraph() local
872const std::vector<tensor::TensorPtr> &output_tensors, size_t *output_position, in ConstructOutputs()
/third_party/ffmpeg/libavfilter/dnn/
Ddnn_backend_tf.c721 TF_Tensor **output_tensors; in execute_model_tf() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc1340 auto output_tensors = TransformVectorRefToMultiTensor(op_outputs); in HandleOpOutputs() local