Searched refs:input_map_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.h | 43 : input_map_(input_map), in OpAdapterImpl()
98 const std::unordered_map<int, InputDesc> &input_map_;
118 … : impl_(std::make_shared<OpAdapterImpl>(input_map_, dyn_input_map_, output_map_, dyn_output_map_, in OpAdapter()
125 … impl_(std::make_shared<OpAdapterImpl>(input_map_, dyn_input_map_, output_map_, dyn_output_map_, in OpAdapter()
186 const std::unordered_map<int, InputDesc> &getInputMap() override { return input_map_; } in getInputMap()
426 static const std::unordered_map<int, InputDesc> input_map_; variable
443 const std::unordered_map<int, InputDesc> OpAdapter<T>::input_map_;
|
| D | op_adapter.cc | 142 auto it = input_map_.find(index); in SetNormalOpInput()
143 if (input != nullptr && it != input_map_.end()) { in SetNormalOpInput()
184 auto it = input_map_.find(index); in SetNormalOpInput()
185 if ((handle.op != nullptr) && (it != input_map_.end())) { in SetNormalOpInput()
413 auto it = input_map_.find(i); in UpdateNormalOpInputDesc()
414 if (it != input_map_.end()) { in UpdateNormalOpInputDesc()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_session.cc | 303 MS_ASSERT(this->input_map_.empty()); in InitGraphInputMap()
311 … MS_ASSERT(this->input_map_.find(in_node->name_ + std::to_string(i)) == this->input_map_.end()); in InitGraphInputMap()
330 this->input_map_[tensor_name] = in_tensor; in InitGraphInputMap()
332 this->input_map_[in_tensor->tensor_name()] = in_tensor; in InitGraphInputMap()
840 input_map_.clear(); in ~LiteSession()
860 auto ret = input_map_.find(name); in GetInputsByTensorName()
861 if (ret == input_map_.end()) { in GetInputsByTensorName()
|
| D | lite_session.h | 147 std::unordered_map<std::string, mindspore::tensor::MSTensor *> input_map_; variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | op_declare_macro.h | 31 const std::unordered_map<int, InputDesc> OpAdapter<T>::input_map_; \
61 const std::unordered_map<int, InputDesc> OpAdapter<T>::input_map_
|