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.cc73 computations_visited_in_backward_pass_.insert( in DetermineFusionComputationPrecision()
198 computations_visited_in_backward_pass_.insert(body); in DetermineWhileComputationsPrecision()
205 computations_visited_in_backward_pass_.insert(condition); in DetermineWhileComputationsPrecision()
231 computations_visited_in_backward_pass_.insert(branch); in DetermineConditionalComputationsPrecision()
838 computations_visited_in_backward_pass_.clear(); in Run()
898 if (ContainsKey(computations_visited_in_backward_pass_, *comp_it)) { in Run()
906 computations_visited_in_backward_pass_.insert(*comp_it); in Run()
Dbfloat16_propagation.h126 computations_visited_in_backward_pass_; variable