Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc364 const string kOnlyArgOrOcInputAttrName = "_xla_only_arg_or_oc_input"; in MoveHeadOutsideCompilationToHost() local
375 !HasNodeAttr(e->src()->def(), kOnlyArgOrOcInputAttrName))) { in MoveHeadOutsideCompilationToHost()
383 n->AddAttr(kOnlyArgOrOcInputAttrName, true); in MoveHeadOutsideCompilationToHost()
400 !HasNodeAttr(e->dst()->def(), kOnlyArgOrOcInputAttrName)) { in MoveHeadOutsideCompilationToHost()