Searched refs:kXlaTokenInputNodesAttrName (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 23 const char kXlaTokenInputNodesAttrName[] = "_xla_token_input_nodes"; variable
86 if (!GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, in CalculateTokenInputsForOutputToken()
116 if (GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, &token_input_nodes) in HasSideEffectingNodes()
|
| D | side_effect_util.h | 32 extern const char kXlaTokenInputNodesAttrName[];
|
| D | graph_compiler.cc | 260 func.attr().find(kXlaTokenInputNodesAttrName) != func.attr().end(); in CompileFunctionalNode()
285 kXlaTokenInputNodesAttrName, in CompileFunctionalNode()
|
| D | xla_compiler_test.cc | 1512 AddNodeAttr(kXlaTokenInputNodesAttrName, in TEST_F()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 119 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in HostComputeOp()
122 kXlaTokenInputNodesAttrName, " attr")); in HostComputeOp()
395 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in SendToHostOp()
398 kXlaTokenInputNodesAttrName, " attr")); in SendToHostOp()
450 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in RecvFromHostOp()
453 kXlaTokenInputNodesAttrName, " attr")); in RecvFromHostOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 354 host_compute_builder.Attr(kXlaTokenInputNodesAttrName, xla_token_input_nodes); in BuildXlaHostComputeNodeDef()
1330 send_pred_builder.Attr(kXlaTokenInputNodesAttrName, in BuildSendIfPredNode()
1513 send_loop_cond_builder.Attr(kXlaTokenInputNodesAttrName, in AddSendLoopPredToLoopCond()
1874 replace->AddAttr(kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForFuncCallNode()
1954 n->AddAttr(kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForIfNode()
2065 n->AddAttr(kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForWhileNode()
2419 kXlaTokenInputNodesAttrName, in ExtractOutsideCompilationForFunction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | case_op.cc | 34 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaCaseOp()
|
| D | if_op.cc | 39 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaIfOp()
|
| D | while_op.cc | 278 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaWhileOp()
|