Home
last modified time | relevance | path

Searched refs:base_ptrs_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc164 InsertOrDie(&base_ptrs_, &hlo, ShapeTree<llvm::Value*>(hlo_shape, nullptr)); in BindHloToIrValue()
166 *(base_ptrs_[&hlo].mutable_element(shape_index)) = typed_ir_value; in BindHloToIrValue()
218 for (auto& key_value : base_ptrs_) { in UnbindAllLocalIrValues()
226 base_ptrs_.erase(hlo_to_unbind); in UnbindAllLocalIrValues()
237 if (base_ptrs_.empty()) { in ToString()
244 base_ptrs_.begin()->first->GetModule()->MakeComputationPostOrder()) { in ToString()
248 auto it = base_ptrs_.find(instr); in ToString()
249 if (it == base_ptrs_.end()) { in ToString()
Dhlo_to_ir_bindings.h63 return base_ptrs_.contains(&hlo); in BoundToIrValue()
73 auto it = base_ptrs_.find(&hlo);
74 CHECK(it != base_ptrs_.end()) << hlo.ToString();
113 base_ptrs_; variable