Home
last modified time | relevance | path

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

12345

/third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/
Dcommon_pybind.h31 std::vector<MSTensor> tensors; in MSTensorPtrToMSTensor() local
42 static inline std::vector<MSTensorPtr> MSTensorToMSTensorPtr(const std::vector<MSTensor> &tensors) { in MSTensorToMSTensorPtr()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dgroup_convolution_creator.cc107 void GroupConvCreator::CopyQuantParam(const std::vector<lite::Tensor *> *tensors) { in CopyQuantParam()
129 int GroupConvCreator::NewInputTensor(std::vector<lite::Tensor *> *tensors) { in NewInputTensor()
140 int GroupConvCreator::NewOutputTensor(std::vector<lite::Tensor *> *tensors, const lite::Tensor *out… in NewOutputTensor()
154 int GroupConvCreator::NewConstTensor(std::vector<lite::Tensor *> *tensors, int group_id) { in NewConstTensor()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Dformat_pass.cc74 int FormatOptimize::RunPass(kernel::SubGraphKernel *graph, std::vector<Tensor *> *tensors) { in RunPass()
89 std::vector<mindspore::lite::Tensor *> *tensors, mindspore::Format graph_format, in DoFormatPass()
109std::vector<mindspore::lite::Tensor *> *tensors, mindspore::Format graph_format, in RuntimeFormatPass()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dtensor_utils.cc113 std::vector<mindspore::tensor::Tensor> tensors; in MSTensorToTensor() local
135 std::vector<MSTensor> TensorUtils::TensorToMSTensor(std::vector<mindspore::tensor::Tensor> tensors, in TensorToMSTensor()
149 const std::vector<mindspore::tensor::Tensor> &tensors) { in TensorToTensorPtr()
164 std::vector<mindspore::tensor::Tensor> tensors; in TensorPtrToTensor() local
222 const std::vector<infer::abstract::Tensor *> &tensors) { in GetTensorListShapes()
237 bool AbstractTensorUtils::SetTensorListShapse(const std::vector<infer::abstract::Tensor *> &tensors, in SetTensorListShapse()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/somas/
Dsomas_solver_pre.cc33 auto tensors = *pTensors; in CheckTensors() local
57 auto &tensors = *pTensors; in AddContiguousInfoInMap() local
92 vector<TensorsDescMap> SomasSolverPre::CreateTensorsMaps(const TensorsDescMap &tensors, size_t tota… in CreateTensorsMaps() argument
132 TensorsDescMap &tensors = *ptensors; in Solving() local
197 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-src/source/mindspore/lite/src/litert/kernel/cpu/control/
Dtensorlist_setitem.cc112 auto tensors = output0_->tensors(); in Run() local
137 auto tensors = output0_->tensors(); in Run() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_concat.cc50 auto tensors = std::make_shared<abstract::AbstractTuple>(inputs); in TEST_P() local
99 auto tensors = std::make_shared<abstract::AbstractTuple>(inputs); in TEST_P() local
149 auto tensors = std::make_shared<abstract::AbstractTuple>(input_elements); in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/
Druntime_pass_tests.cc36 …4hw4PassConstruct(std::vector<kernel::KernelExec *> *kernels, std::vector<lite::Tensor *> *tensors, in Nc4hw4PassConstruct()
140 …rmC4PassConstruct(std::vector<kernel::KernelExec *> *kernels, std::vector<lite::Tensor *> *tensors, in ConvNormC4PassConstruct()
247 std::vector<lite::Tensor *> tensors; in TEST_F() local
271 std::vector<lite::Tensor *> tensors; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/
Dtensorlist.h112 std::vector<Tensor *> tensors() { return tensors_; } in tensors() function
114 void set_tensors(const std::vector<Tensor *> &tensors) { this->tensors_ = tensors; } in set_tensors()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dhost_tensor_queue.h36 void Push(const std::vector<TensorPtr> &tensors) { buffers_.push(tensors); } in Push()
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/cloud_infer/device_example_cpp/
Dexample.cc69 int SetHostData(std::vector<mindspore::MSTensor> tensors, const std::vector<uint8_t *> &host_data_b… in SetHostData()
77 int SetDeviceData(std::vector<mindspore::MSTensor> tensors, const std::vector<uint8_t *> &host_data… in SetDeviceData()
103 int SetOutputHostData(std::vector<mindspore::MSTensor> tensors, std::vector<uint8_t *> *host_buffer… in SetOutputHostData()
119 int SetOutputDeviceData(std::vector<mindspore::MSTensor> tensors, std::vector<void *> *device_buffe… in SetOutputDeviceData()
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/cloud_infer/ascend_ge_distributed_java/src/main/java/com/mindspore/lite/demo/
DMain.java107 List<MSTensor> tensors = models.get(mdlIndex).getInputs(); in run() local
180 private static void freeTensorBuffer(List<MSTensor> tensors) { in freeTensorBuffer()
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/cpp/device_example_cpp/
Dexample.cc71 int SetHostData(std::vector<mindspore::MSTensor> tensors, const std::vector<uint8_t *> &host_data_b… in SetHostData()
79 int SetDeviceData(std::vector<mindspore::MSTensor> tensors, const std::vector<uint8_t *> &host_data… in SetDeviceData()
105 int SetOutputHostData(std::vector<mindspore::MSTensor> tensors, std::vector<uint8_t *> *host_buffer… in SetOutputHostData()
121 int SetOutputDeviceData(std::vector<mindspore::MSTensor> tensors, std::vector<void *> *device_buffe… in SetOutputDeviceData()
/third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/java/com/mindspore/
DModelParallelRunner.java173 List<MSTensor> tensors = new ArrayList<>(); in getInputs() local
195 List<MSTensor> tensors = new ArrayList<>(); in getOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/
Dconcat.cc31 void ConcatCpuCustomize(const std::shared_ptr<OpRunner> &op, const ValueTuplePtr &tensors, const In… in ConcatCpuCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dragged_tensor_to_tensor.cc44 abstract::BaseShapePtrList tensors; in RaggedTensorToTensorInferShape() local
121 TypePtrList tensors; in RaggedTensorToTensorInferType() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/
Dtest_addn.py33 def get_output(*tensors):
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Druntime_pass.cc43 void Nc4hw4PassReplace(std::vector<kernel::KernelExec *> *kernels, std::vector<Tensor *> *tensors, … in Nc4hw4PassReplace()
177 void Nc4hw4PassAct(std::vector<kernel::KernelExec *> *kernels, std::vector<Tensor *> *tensors, int … in Nc4hw4PassAct()
327 STATUS RuntimePass(std::vector<kernel::KernelExec *> *subgraphs, std::vector<Tensor *> *tensors) { in RuntimePass()
Druntime_packed_node_pass.cc43 void PackedNodePass::Run(Model *model, const std::vector<Tensor *> &tensors) { in Run()
261 int PackedMatmulKernelExec(kernel::KernelExec *kernel_exec, const std::vector<Tensor *> &tensors) { in PackedMatmulKernelExec()
333 int PackKernelExec(kernel::KernelExec *kernel_exec, const std::vector<Tensor *> &tensors) { in PackKernelExec()
/third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/
Dtest_cat_dyn.py29 def construct(self, tensors, axis): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/
Dconcat_aclnn_kernel.cc30 std::vector<KernelTensor *> tensors(inputs.begin(), last_element); in GetConcatRealInputs() local
Dstack_ext_aclnn_kernel.cc30 std::vector<KernelTensor *> tensors(inputs.begin(), last_element); in GetStackRealInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/internal/
Dreshape.cc34 …auto calc_size_list = [](const std::vector<KernelTensor *> &tensors, std::vector<size_t> *list_ptr… in Resize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dtensorlist_c_utils.c39 TensorC *tensors = (TensorC *)(tensor_list->tensors_ + tensor_list->element_num_); in MallocTensorListData() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/
Dhost_kernel_mod.cc51 …auto calc_size_list = [](const std::vector<KernelTensor *> &tensors, std::vector<size_t> *list_ptr… in Resize()

12345