Home
last modified time | relevance | path

Searched defs:origin_tensor (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/npu/
Dnpu_subgraph.cc51 auto origin_tensor = inputs_[index]; in set_input() local
65 auto origin_tensor = outputs_[index]; in set_output() local
Dnpu_graph.cc38 auto origin_tensor = this->inputs_[index]; in set_input() local
51 auto origin_tensor = this->outputs_[index]; in set_output() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc241 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(weight_param->default_param()); in SplitWeights() local
294 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(old_bias_param->default_param()); in PopulateBiasNode() local
Dconv_bn_fusion.cc124 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(origin_param); in CalEstimatedData() local
Dtflite_lstm_cell_fusion.cc440 auto origin_tensor = std::dynamic_pointer_cast<tensor::Tensor>(param_t->default_param()); in GetConcatedParam() local
/third_party/mindspore/mindspore/lite/micro/coder/
Dgraph.cc65 schema::Tensor *origin_tensor = model_->all_tensors_.at(i); in ConvertTensors() local
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_session.cc295 auto *origin_tensor = restored_origin_tensor.first; in RestoreTensorData() local
/third_party/mindspore/mindspore/lite/src/
Dscheduler.cc842 auto *origin_tensor = restored_origin_tensor.first; in RestoreTensorData() local