Searched defs:new_tensor (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/ |
| D | opencl_subgraph.cc | 40 … lite::Tensor *new_tensor, kernel::LiteKernel *in_convert_op, in ReplaceOutTensorAndKernelToConvert()
90 auto *new_tensor = new (std::nothrow) in GenToFormatOp() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | group_convolution_creator.cc | 65 auto new_tensor = in CreateConstTensor() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | random_data_op.cc | 280 std::shared_ptr<Tensor> new_tensor = nullptr; in CreateRandomRow() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 50 auto new_tensor = std::make_shared<tensor::Tensor>(type_id, cache_shape); in AddCacheParameters() local
248 auto new_tensor = std::make_shared<tensor::Tensor>(type_id, host_shape); in InitHashMap() local
267 auto new_tensor = std::make_shared<tensor::Tensor>(type_id, host_shape); in InitStep() local
534 auto new_tensor = std::make_shared<tensor::Tensor>(ori_input_type_id, input_shape); in CreateOutputNodeParam() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_parse_graph.cc | 32 TensorParam new_tensor; in MakeTensor() local
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_mindrt.cc | 77 …eOpActor::ReplaceNodeInTensor(kernel::LiteKernel *kernel, Tensor *old_tensor, Tensor *new_tensor) { in ReplaceNodeInTensor()
129 …Tensor *new_tensor = new Tensor(new_data_type, old_tensor->shape(), old_tensor->format(), old_tens… in IsolateInputData() local
|
| D | lite_session.cc | 418 Tensor *new_tensor = in IsolateOutputTensor() local
|
| D | scheduler.cc | 1717 auto new_tensor = Tensor::CopyTensor(*tensor, false); in CopyTensorList() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 193 std::shared_ptr<Tensor> new_tensor; in BatchRows() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | special_op_eliminate.h | 543 … auto new_tensor = std::make_shared<tensor::Tensor>(tensor->Dtype()->type_id(), tensor->shape()); in OperatorHandle3() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 555 …tensor::TensorPtr new_tensor = std::make_shared<tensor::Tensor>(kNumberTypeFloat32, new_tensor_sha… in SyncEmbeddingTables() local
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 718 void UpdateTensorInfo(const tensor::TensorPtr &new_tensor, const std::vector<tensor::TensorPtr> &pr… in UpdateTensorInfo()
1755 const auto &new_tensor = all_op_tensors[i]; in UpdateForwardTensorInfoInBpropGraph() local
1898 …auto new_tensor = std::make_shared<tensor::Tensor>(tensor->data_type(), tensor->shape(), tensor->d… in RunOpInVM() local
|