Searched refs:SetNodeToken (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 417 Status SetNodeToken(const string& node_name, const xla::XlaOp& op);
|
| D | graph_compiler.cc | 300 TF_RETURN_IF_ERROR(compiler->SetNodeToken( in CompileFunctionalNode()
|
| D | xla_compiler.cc | 1109 SetNodeToken(kXlaTokenArgNodeName, arg_expression.handle())); in CompileGraph()
1123 SetNodeToken(kXlaTokenArgNodeName, xla::CreateToken(&builder))); in CompileGraph()
1307 Status XlaCompiler::SetNodeToken(const string& node_name, in SetNodeToken() function in tensorflow::XlaCompiler
|
| D | xla_compiler_test.cc | 1432 OP_REQUIRES_OK(ctx, ctx->compiler()->SetNodeToken( in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_op.cc | 255 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| D | case_op.cc | 267 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| D | while_op.cc | 549 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|