Searched refs:SetNodeToken (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 239 ctx, ctx->compiler()->SetNodeToken(original_node_name_, token_output)); in Compile()
432 compiler->SetNodeToken(original_node_name_, output_token)); in Compile()
485 ctx, compiler->SetNodeToken(name(), xla::GetTupleElement(result, 1))); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 289 Status SetNodeToken(const string& node_name, const xla::XlaOp& op);
|
| D | graph_compiler.cc | 327 TF_RETURN_IF_ERROR(compiler->SetNodeToken( in CompileFunctionalNode()
|
| D | xla_compiler.cc | 1346 SetNodeToken(kXlaTokenArgNodeName, arg_expression.handle())); in CompileGraph()
1360 SetNodeToken(kXlaTokenArgNodeName, xla::CreateToken(&builder))); in CompileGraph()
1547 Status XlaCompiler::SetNodeToken(const string& node_name, in SetNodeToken() function in tensorflow::XlaCompiler
|
| D | xla_compiler_test.cc | 1498 OP_REQUIRES_OK(ctx, ctx->compiler()->SetNodeToken( in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | case_op.cc | 344 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| D | if_op.cc | 329 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| D | while_op.cc | 602 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|