Home
last modified time | relevance | path

Searched defs:src_tensor (Results 1 – 25 of 42) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dtensor.cc153 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorGetData() local
166 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorSetDataType() local
185 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorGetDataType() local
205 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorGetDataSize() local
226 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorGetElementNum() local
247 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorGetDimension() local
263 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorSetShape() local
283 auto src_tensor = GetSrcPtr<TensorPtr>(res_mgr, tensor); in MSTensorGetShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnrt/checker/
Dprimitive_check.cc11 bool NeedBitUppackCheck(const schema::Tensor &src_tensor) { in NeedBitUppackCheck()
24 int DecompressTensor(const schema::Tensor &src_tensor) { in DecompressTensor()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dmove_to.cc26 bool MoveToD2H(const tensor::TensorPtr &src_tensor, const DeviceAddressPtr &src_device_ptr, in MoveToD2H()
44 void MoveToH2D(const tensor::TensorPtr &src_tensor, const DeviceAddressPtr &src_device_ptr, in MoveToH2D()
63 void MoveTo(const tensor::TensorPtr &src_tensor, const tensor::TensorPtr &dst_tensor, const std::st… in MoveTo()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/device/
Dmove_to_test.cc53 auto src_tensor = std::make_shared<tensor::Tensor>(input, type); in TEST_F() local
73 auto src_tensor = std::make_shared<tensor::Tensor>(input, type); in TEST_F() local
93 auto src_tensor = std::make_shared<tensor::Tensor>(input, type); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dweight_decoder.cc114 int WeightDecoder::DecodeHuffmanCode(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tenso… 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()
472 bool NeedBitUppackCheck(const SchemaTensorWrapper &src_tensor) { in NeedBitUppackCheck()
519 int WeightDecoder::DecompressTensor(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tensor… in DecompressTensor()
Dmindrt_executor.cc267 auto src_tensor = tensor_map.first; in TransferGraphOutput() local
308 auto src_tensor = tensor_map.first; in FreeOutputTensor() local
Dlite_session.cc128 void LiteSession::ConvertTensorsQuantParam(const schema::Tensor *src_tensor, lite::Tensor *dst_tens… in ConvertTensorsQuantParam()
167 auto src_tensor = model->GetSchemaTensor(tensor_index); in ConvertTensorsData() local
208 lite::Tensor *LiteSession::ConvertTensor(const schema::Tensor &src_tensor) { in ConvertTensor()
265 auto *src_tensor = model->graph_.all_tensors_[i]; in ConvertTensors() local
1424 auto src_tensor = tensor_map.first; in SynIsolateInOutputDataType() local
1431 auto src_tensor = tensor_map.first; in SynIsolateInOutputDataType() local
Dlite_mindrt.cc178 auto src_tensor = map.second; in ResizeGraphInput() local
354 auto src_tensor = inputs_data_[i]; in AssignInputData() local
Dcache_session.cc102 auto *src_tensor = model->graph_.all_tensors_[i]; in ConvertInOutTensors() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/
Didentity_kernel.cc27 auto src_tensor = in_tensors()[i]; in Run() local
66 auto src_tensor = in_tensors()[i]; in InferShape() 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()
82 auto *src_tensor = *src_iter; in MoveData() local
/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()
489 auto src_tensor = src_tensorlist->tensors()[i]; in CastTensorListTensorData() local
519 auto src_tensor = src_tensorlist->tensors()[i]; in MoveTensorListTensorData() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/view/
Dcopy_strides_calc.cc29 auto src_tensor = inputs[1]->cast<tensor::BaseTensorPtr>(); in CopyWithSliceCalc() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dscatter_add_ext.cc31 … const BaseTensorPtr &index_tensor, const BaseTensorPtr &src_tensor) { in ScatterAddExtAscendCustomize()
Dscatter.cc28const BaseTensorPtr &src_tensor, const Int64ImmPtr &reduce) { in ScatterAscendCustomize()
/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/control_flow/actor/
Dentrance_actor.cc55 auto src_tensor = inputs_data_[i]; in InitInputData() local
Dexit_actor.cc51 auto src_tensor = inputs_data_[i]; 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/src/
Dtensor.cc62 int Tensor::CopyTensorData(const Tensor &src_tensor, Tensor *dst_tensor) { in CopyTensorData()
85 Tensor *Tensor::CopyTensor(const Tensor &src_tensor, bool copy_data, AllocatorPtr allocator) { in CopyTensor()
Dtensorlist.cc148 int TensorList::SetTensor(int index, const Tensor *src_tensor) { in SetTensor()
/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/
Dcast_base.cc160 auto src_tensor = src_value->cast<tensor::BaseTensorPtr>(); in TensorToDstDtypeValue() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dsparse_reshape_fp32.cc40 int SparseReshapeCPUKernel::SoftCopyInputToOutput(lite::Tensor *src_tensor, lite::Tensor *dst_tenso… in SoftCopyInputToOutput()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/
Dmindir_model.cc48 auto src_tensor = this->all_mindir_tensors_[i]; in ConvertTensors() local

12