Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc222 const AttrValueMap* current_function_attr, in FindTPUCompileNodes() argument
261 if (current_function_attr) { in FindTPUCompileNodes()
263 *current_function_attr; in FindTPUCompileNodes()
390 const AttrValueMap* current_function_attr, in DetectHostTrainingLoop() argument
435 FindTPUCompileNodes(current_function_name, current_function_attr, frames, in DetectHostTrainingLoop()
Dhost_training_loop_optimization_util.h67 const AttrValueMap* current_function_attr,