Searched refs:kXlaOriginalOutsideCompilationNodeName (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 34 const char kXlaOriginalOutsideCompilationNodeName[] = variable
100 kXlaOriginalOutsideCompilationNodeName, in CalculateTokenInputsForOutputToken()
|
| D | side_effect_util.h | 48 extern const char kXlaOriginalOutsideCompilationNodeName[];
|
| D | xla_compiler_test.cc | 1514 AddNodeAttr(kXlaOriginalOutsideCompilationNodeName, side_effecting_op.name(), in TEST_F()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 123 OP_REQUIRES_OK(ctx, ctx->GetAttr(kXlaOriginalOutsideCompilationNodeName, in HostComputeOp()
399 OP_REQUIRES_OK(ctx, ctx->GetAttr(kXlaOriginalOutsideCompilationNodeName, in SendToHostOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 321 host_compute_builder.Attr(kXlaOriginalOutsideCompilationNodeName, in BuildXlaHostComputeNodeDef()
1332 send_pred_builder.Attr(kXlaOriginalOutsideCompilationNodeName, name); in BuildSendIfPredNode()
1515 send_loop_cond_builder.Attr(kXlaOriginalOutsideCompilationNodeName, in AddSendLoopPredToLoopCond()
1876 replace->AddAttr(kXlaOriginalOutsideCompilationNodeName, replace->name()); in ExtractOutsideCompilationForFuncCallNode()
1943 n->AddAttr(kXlaOriginalOutsideCompilationNodeName, n->name()); in ExtractOutsideCompilationForIfNode()
2057 n->AddAttr(kXlaOriginalOutsideCompilationNodeName, n->name()); in ExtractOutsideCompilationForWhileNode()
|