Searched refs:output_map_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_converter_context.cc | 118 output_map_[input_node][0] = LayerInput(weight, tensor->shape()); in InitInputTable()
145 output_map_[value_node][i] = LayerInput(weight, tensor->shape()); in InitValueNodeTable()
163 output_map_[node][tensor_index] = LayerInput(nv_tensors[tensor_index], shape); in StoreLayerOutput()
186 output_map_[node][0] = LayerInput(tensor, shape); in LoadInputOnDemand()
187 return &output_map_[node][0]; in LoadInputOnDemand()
194 auto node_iter = output_map_.find(item.first); in LoadLayerInput()
195 if (node_iter == output_map_.end()) { in LoadLayerInput()
|
| D | trt_converter_context.h | 93 std::unordered_map<AnfNodePtr, std::unordered_map<size_t, LayerInput>> output_map_; variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.cc | 263 if (!dyn_output_map_.empty() && !output_map_.empty()) { in getNormalOutput()
267 auto it = output_map_.find(index); in getNormalOutput()
268 if (it != output_map_.end()) { in getNormalOutput()
299 if (output_map_.empty()) { in UpdateSingleOutputDesc()
303 output_map_.begin()->second.update_out_desc(op, *desc); in UpdateSingleOutputDesc()
343 output_size = output_map_.size(); in UpdateMultiOutputDesc()
371 auto it = output_map_.find(i); in UpdateMultiOutputDesc()
372 if (it != output_map_.end()) { in UpdateMultiOutputDesc()
|
| D | op_adapter.h | 45 output_map_(output_map), in OpAdapterImpl()
100 const std::unordered_map<int, OutputDesc> &output_map_; variable
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()
189 const std::unordered_map<int, OutputDesc> &getOutputMap() override { return output_map_; } in getOutputMap()
428 static const std::unordered_map<int, OutputDesc> output_map_; variable
447 const std::unordered_map<int, OutputDesc> OpAdapter<T>::output_map_;
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | op_declare_macro.h | 37 const std::unordered_map<int, OutputDesc> OpAdapter<T>::output_map_;
135 const std::unordered_map<int, OutputDesc> OpAdapter<T>::output_map_
|