Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation.cc71 computations_visited_in_backward_pass_.insert( in DetermineFusionComputationPrecision()
196 computations_visited_in_backward_pass_.insert(body); in DetermineWhileComputationsPrecision()
203 computations_visited_in_backward_pass_.insert(condition); in DetermineWhileComputationsPrecision()
718 computations_visited_in_backward_pass_.clear(); in Run()
778 if (ContainsKey(computations_visited_in_backward_pass_, *comp_it)) { in Run()
786 computations_visited_in_backward_pass_.insert(*comp_it); in Run()
Dbfloat16_propagation.h121 computations_visited_in_backward_pass_; variable