Searched refs:handle_to_index_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 146 auto it = handle_to_index_.find(op.handle()); in GetShapePtr()
147 if (it == handle_to_index_.end()) { in GetShapePtr()
436 this->handle_to_index_.clear(); in Build()
2789 handle_to_index_[handle] = instructions_.size(); in AddInstruction()
2851 handle_to_index_, instructions_, op.builder_, this, op.handle()); in LookUpInstruction()
2857 handle_to_index_, instructions_, handle); in LookUpInstructionByHandle()
2864 handle_to_index_, instructions_, op.builder_, this, op.handle()); in LookUpMutableInstruction()
2870 handle_to_index_, instructions_, handle); in LookUpMutableInstructionByHandle()
|
| D | xla_builder.h | 722 absl::flat_hash_map<int64, int64> handle_to_index_; variable
|