Home
last modified time | relevance | path

Searched defs:dst_tensor (Results 1 – 25 of 32) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/device/
Dmove_to_test.cc54 auto dst_tensor = std::make_shared<tensor::Tensor>(src_tensor->data_type(), src_tensor->shape()); in TEST_F() local
74 auto dst_tensor = std::make_shared<tensor::Tensor>(src_tensor->data_type(), src_tensor->shape()); in TEST_F() local
96 auto dst_tensor = std::make_shared<tensor::Tensor>(src_tensor->data_type(), src_tensor->shape()); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/
Didentity_kernel.cc28 auto dst_tensor = out_tensors()[i]; in Run() local
67 auto dst_tensor = out_tensors()[i]; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dweight_decoder.cc114 …WeightDecoder::DecodeHuffmanCode(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tensor) { in DecodeHuffmanCode()
143 int WeightDecoder::UnPackToInt(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tensor) { in UnPackToInt()
178 int WeightDecoder::UnPack(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tensor) { in UnPack()
203 STATUS WeightDecoder::SparseDecompress(const SchemaTensorWrapper &src_tensor, Tensor *dst_tensor) { in SparseDecompress()
310 STATUS WeightDecoder::IndexingDecompress(const SchemaTensorWrapper &src_tensor, Tensor *dst_tensor)… in IndexingDecompress()
519 int WeightDecoder::DecompressTensor(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tensor in DecompressTensor()
Dmindrt_executor.cc266 auto dst_tensor = tensor_map.second; in TransferGraphOutput() local
309 auto dst_tensor = tensor_map.second; in FreeOutputTensor() local
Dlite_session.cc114 int LiteSession::CheckTensorValid(lite::Tensor *dst_tensor) { in CheckTensorValid()
128 …LiteSession::ConvertTensorsQuantParam(const schema::Tensor *src_tensor, lite::Tensor *dst_tensor) { in ConvertTensorsQuantParam()
164 …::ConvertTensorsData(const lite::LiteModel *model, size_t tensor_index, lite::Tensor *dst_tensor) { in ConvertTensorsData()
232 lite::Tensor *dst_tensor = nullptr; in ConvertTensor() local
270 auto *dst_tensor = ConvertTensor(*src_tensor); in ConvertTensors() local
1423 auto dst_tensor = tensor_map.second; in SynIsolateInOutputDataType() local
1430 auto dst_tensor = tensor_map.second; in SynIsolateInOutputDataType() local
Dcache_session.cc111 auto *dst_tensor = ConvertTensor(*src_tensor); in ConvertInOutTensors() local
Druntime_packed_node_pass.cc268 auto dst_tensor = kernel_exec->in_tensors()[SECOND_INPUT]; in PackedMatmulKernelExec() local
Dlite_mindrt.cc353 auto dst_tensor = kernel_->in_tensors()[i]; in AssignInputData() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dmove_to.cc27 const tensor::TensorPtr &dst_tensor, bool blocking) { in MoveToD2H()
63 void MoveTo(const tensor::TensorPtr &src_tensor, const tensor::TensorPtr &dst_tensor, const std::st… in MoveTo()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/
Dtensor_util.cc251 int MoveCommonTensorData(Tensor *dst_tensor, Tensor *src_tensor) { in MoveCommonTensorData()
274 int MoveTensorData(Tensor *dst_tensor, Tensor *src_tensor) { in MoveTensorData()
290 void SetCommonTensorData(Tensor *dst_tensor, Tensor *src_tensor) { in SetCommonTensorData()
295 int SetTensorData(Tensor *dst_tensor, Tensor *src_tensor) { in SetTensorData()
367 bool NeedCastData(Tensor *dst_tensor, Tensor *src_tensor) { in NeedCastData()
490 auto dst_tensor = dst_tensorlist->tensors()[i]; in CastTensorListTensorData() local
520 auto dst_tensor = dst_tensorlist->tensors()[i]; in MoveTensorListTensorData() local
603 auto dst_tensor = new (std::nothrow) TensorList(shape, std::vector<int>(), src_category); in CreateTensorList() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dselect.cc38 int CopyTensorData(lite::Tensor *dst_tensor, const lite::Tensor *src_tensor) { in CopyTensorData()
81 auto *dst_tensor = *dst_iter; in MoveData() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Dpass_utils.cc47 bool SetShape(const Tensor *src_tensor, Tensor *dst_tensor) { in SetShape()
67 bool SetShape4D(const Tensor *src_tensor, Tensor *dst_tensor) { in SetShape4D()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/
Dmindir_model.cc49 auto *dst_tensor = ConvertTensor(src_tensor); in ConvertTensors() local
182 int MindirModel::CheckTensorValid(lite::Tensor *dst_tensor) { in CheckTensorValid()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/actor/
Dentrance_actor.cc54 auto dst_tensor = kernel_->out_tensors()[i + 1]; in InitInputData() local
Dexit_actor.cc50 auto dst_tensor = kernel_->out_tensors()[i - 1]; in InitInputData() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/bolt/
Dbolt_tensor_utils.cc86 int LiteTensor2BoltTensor(const lite::Tensor *src_tensor, BoltTensor *dst_tensor) { in LiteTensor2BoltTensor()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dfse_decoder.cc164 int FSEDecoder::DeCompress(const SchemaTensorWrapper &src_tensor, Tensor *dst_tensor, in DeCompress()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/
Ddo_pyboost_cast.cc44 auto dst_tensor = TensorToDstDtypeValue(t, dst_type.first); in DoAutoCast() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dsparse_reshape_fp32.cc40 …SparseReshapeCPUKernel::SoftCopyInputToOutput(lite::Tensor *src_tensor, lite::Tensor *dst_tensor) { in SoftCopyInputToOutput()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/
Dtensorlist.cc162 auto dst_tensor = this->tensors_[index]; in SetTensor() local
Dtensor.cc62 int Tensor::CopyTensorData(const Tensor &src_tensor, Tensor *dst_tensor) { in CopyTensorData()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Dcompile_result_builder.cc121 void CompileResultBuilder::ReplaceTensor(InferTensor *dst_tensor, const InferTensor *src_tensor) { in ReplaceTensor()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dtensor_util.cc177 …rTFromTensorInfo(const tensor::TensorPtr &src_tensor, std::unique_ptr<schema::TensorT> *dst_tensor in UpdateTensorTFromTensorInfo()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dsub_graph_kernel.h237 auto dst_tensor = inputs[1]; in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_device_res_manager.cc532 …eDeviceResManager::MoveTo(const tensor::TensorPtr &src_tensor, const tensor::TensorPtr &dst_tensor, in MoveTo()

12