Searched refs:kXlaTokenInputNodesAttrName (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 23 const char kXlaTokenInputNodesAttrName[] = "_xla_token_input_nodes"; variable
78 if (!GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, in CalculateTokenInputsForOutputToken()
104 if (GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, &token_input_nodes) in HasSideEffectingNodes()
|
| D | side_effect_util.h | 32 extern const char kXlaTokenInputNodesAttrName[];
|
| D | graph_compiler.cc | 253 func.attr().find(kXlaTokenInputNodesAttrName) != func.attr().end(); in CompileFunctionalNode()
274 kXlaTokenInputNodesAttrName, in CompileFunctionalNode()
|
| D | xla_compiler_test.cc | 1446 AddNodeAttr(kXlaTokenInputNodesAttrName, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 313 host_compute_builder.Attr(kXlaTokenInputNodesAttrName, in BuildXlaHostComputeNodeDef()
811 send_pred_builder.Attr(kXlaTokenInputNodesAttrName, in BuildSendIfPredNode()
1001 send_loop_cond_builder.Attr(kXlaTokenInputNodesAttrName, in AddSendLoopPredToLoopCond()
1339 replace->AddAttr(kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1404 n->AddAttr(kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1464 n->AddAttr(kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_op.cc | 37 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaIfOp()
|
| D | case_op.cc | 32 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaCaseOp()
|
| D | while_op.cc | 260 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaWhileOp()
|