Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dgroup_convolution_creator.cc107 void GroupConvCreator::CopyQuantParam(std::vector<lite::Tensor *> *tensors) { in CopyQuantParam()
127 int GroupConvCreator::NewInputTensor(std::vector<lite::Tensor *> *tensors) { in NewInputTensor()
137 int GroupConvCreator::NewOutputTensor(std::vector<lite::Tensor *> *tensors, lite::Tensor *output) { in NewOutputTensor()
149 int GroupConvCreator::NewConstTensor(std::vector<lite::Tensor *> *tensors, int group_id) { in NewConstTensor()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas_solver_pre.cc32 auto tensors = *pTensors; in CheckTensors() local
51 auto &tensors = *pTensors; in AddContiguousInfoInMap() local
86 vector<TensorsDescMap> SomasSolverPre::CreateTensorsMaps(const TensorsDescMap &tensors, size_t tota… in CreateTensorsMaps()
104 TensorsDescMap &tensors = *ptensors; in Solving() local
196 void SomasSolverPre::Log(const session::KernelGraph *graph, const TensorsDescMap &tensors, in Log()
229 …id SomasSolverPre::SolverInputLog(const session::KernelGraph *graph, const TensorsDescMap &tensors, in SolverInputLog()
/third_party/mindspore/mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/model/
DSessionUtil.java50 public static Map<String, float[]> convertTensorToFeatures(List<MSTensor> tensors) { in convertTensorToFeatures()
92 List<MSTensor> tensors = new ArrayList<MSTensor>(featureMaps.size()); in updateFeatures() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/
Dtensorlist_setitem.cc113 auto &tensors = output0_->tensors(); in Run() local
134 auto &tensors = output0_->tensors(); in Run() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc208 const auto &tensors = input_tensors[i]; in PrepareDataForDeviceTensorStore() local
239 const auto &tensors = input_tensors[i]; in PrepareDataForHostTensorQueue() local
283 const auto &tensors = input_tensors[i]; in PrepareDataForStepMode() local
340 std::vector<TensorPtr> tensors; in PrepareDataForValueNodeTensor() local
508 const std::vector<TensorPtr> &tensors, in PrepareDeviceTensorStoreForControlNode()
537 void DataPrepareActor::PrepareHostTensorQueueForControlNode(const std::vector<TensorPtr> &tensors, in PrepareHostTensorQueueForControlNode()
/third_party/mindspore/mindspore/lite/src/
Dtensorlist.h98 std::vector<Tensor *> &tensors() { return tensors_; } in tensors() function
100 void set_tensors(const std::vector<Tensor *> &tensors) { this->tensors_ = tensors; } in set_tensors()
/third_party/mindspore/mindspore/lite/src/common/
Dtensor_util.cc25 int OutputTensor2TensorC(const std::vector<lite::Tensor *> &tensors, std::vector<TensorC *> *tensor… in OutputTensor2TensorC()
237 int CheckTensorsInvalid(const std::vector<Tensor *> &tensors) { in CheckTensorsInvalid()
267 std::vector<mindspore::MSTensor> tensors; in LiteTensorsToMSTensors() local
/third_party/mindspore/mindspore/lite/src/runtime/
Druntime_pass.cc25 void Nc4hw4PassReplace(std::vector<kernel::LiteKernel *> *kernels, std::vector<Tensor *> *tensors, … in Nc4hw4PassReplace()
156 void Nc4hw4PassAct(std::vector<kernel::LiteKernel *> *kernels, std::vector<Tensor *> *tensors, int … in Nc4hw4PassAct()
230 void RuntimePass(std::vector<kernel::LiteKernel *> *subgraphs, std::vector<Tensor *> *tensors) { in RuntimePass()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dhost_tensor_queue.h36 void Push(const std::vector<TensorPtr> &tensors) { buffers_.push(tensors); } in Push()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/
Druntime_pass_tests.cc33 …4hw4PassConstruct(std::vector<kernel::LiteKernel *> *kernels, std::vector<lite::Tensor *> *tensors, in Nc4hw4PassConstruct()
119 …rmC4PassConstruct(std::vector<kernel::LiteKernel *> *kernels, std::vector<lite::Tensor *> *tensors, in ConvNormC4PassConstruct()
207 std::vector<lite::Tensor *> tensors; in TEST_F() local
231 std::vector<lite::Tensor *> tensors; in TEST_F() local
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_addn.py33 def get_output(*tensors):
/third_party/mindspore/third_party/proto/tensorflow/
Dtensor.proto95 repeated TensorProto tensors = 3; field
/third_party/mindspore/mindspore/lite/java/java/app/src/main/java/com/mindspore/lite/
DLiteSession.java146 ArrayList<MSTensor> tensors = new ArrayList<MSTensor>(); in getInputs() local
176 ArrayList<MSTensor> tensors = new ArrayList<>(); in getOutputsByNodeName() local
324 ArrayList<MSTensor> tensors = new ArrayList<MSTensor>(); in getFeaturesMap() local
/third_party/mindspore/mindspore/lite/java/java/linux_x86/src/main/java/com.mindspore.lite/
DLiteSession.java146 ArrayList<MSTensor> tensors = new ArrayList<MSTensor>(); in getInputs() local
176 ArrayList<MSTensor> tensors = new ArrayList<>(); in getOutputsByNodeName() local
324 ArrayList<MSTensor> tensors = new ArrayList<MSTensor>(); in getFeaturesMap() local
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/
Dms_tensor_ref.cc20 VectorRef MSTensorRef::Convert(const std::vector<MSTensor> &tensors) { in Convert()
/third_party/mindspore/mindspore/lite/micro/coder/utils/
Dcoder_utils.cc82 std::string TensorsToString(const std::vector<Tensor *> &tensors, const std::string &is_input) { in TensorsToString()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/
Dopencl_subgraph.cc53 auto tensors = (mem_type == MemType::IMG) ? iv->in_tensors() : iv->out_tensors(); in ReplaceOutTensorAndKernelToConvert() local
153 auto tensors = iv->in_tensors(); in GenToFormatOp() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/custom/
Dcustom_coder.cc66 …:TransformTensors(Serializer *code, std::string array_name, const std::vector<Tensor *> &tensors) { in TransformTensors()
/third_party/mindspore/mindspore/numpy/
Dutils.py107 def _check_input_tensor(*tensors):
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dnode_infershape.cc35 void FreeTensors(std::vector<lite::Tensor *> *tensors) { in FreeTensors()
429 std::vector<lite::Tensor *> *tensors) { in ConvertToLiteTensor()
/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dinfershape_pass.cc118 std::vector<Tensor *> tensors{}; in ConvertTensorList() local
269 …(MetaGraphT *graph, const std::vector<Tensor *> &output_tensors, std::vector<InferTensor> *tensors, in SetDataType()
/third_party/mindspore/mindspore/lite/micro/coder/
Dgraph.cc197 std::vector<Tensor *> tensors = pair.second; in InitInputs() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils.cc278 void TensorValueToTensor(const ValuePtr &value, std::vector<tensor::TensorPtr> *tensors) { in TensorValueToTensor()
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Dgrpc_client.cc144 EventReply GrpcClient::SendTensors(const std::list<TensorProto> &tensors) { in SendTensors()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_converter_context.cc135 std::vector<tensor::TensorPtr> tensors; in InitValueNodeTable() local

12