Searched refs:value_to_index (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | impl.cpp | 598 c10::FastMap<const Value*, uint32_t> value_to_index; in StaticModule() local
599 prepareFunctionsAndConstants(graph_->block(), alias_db, value_to_index); in StaticModule()
606 prepareBlockInfo(graph_->block(), values_index_offset, value_to_index); in StaticModule()
608 prepareStaticNodeInfos(graph_->block(), value_to_index, alias_db); in StaticModule()
619 c10::FastMap<const Value*, uint32_t>& value_to_index) { in prepareBlockInfo() argument
624 value_to_index.emplace(block->inputs()[i], start_idx + i); in prepareBlockInfo()
630 cur_idx += prepareBlockInfo(sub_block, cur_idx, value_to_index); in prepareBlockInfo()
645 value_to_index.emplace(node->outputs()[i], cur_idx + i); in prepareBlockInfo()
653 const auto output_idx = value_to_index.at(output); in prepareBlockInfo()
680 c10::FastMap<const Value*, uint32_t>& value_to_index) { in prepareFunctionsAndConstants() argument
[all …]
|
| D | impl.h | 515 c10::FastMap<const Value*, uint32_t>& value_to_index);
520 c10::FastMap<const Value*, uint32_t>& value_to_index);
530 const c10::FastMap<const Value*, uint32_t>& value_to_index,
|