Searched refs:token_output (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_op.cc | 321 xla::XlaOp token_output = in Compile() local
323 auto shape_or = b->GetShape(token_output); in Compile()
329 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| D | case_op.cc | 336 xla::XlaOp token_output = in Compile() local
338 auto shape_or = b->GetShape(token_output); in Compile()
344 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| D | while_op.cc | 594 xla::XlaOp token_output = in Compile() local
596 auto shape_or = builder->GetShape(token_output); in Compile()
602 OP_REQUIRES_OK(ctx, compiler->SetNodeToken(name(), token_output)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 170 std::unique_ptr<xla::XlaOp> token_output, in BuildComputation() argument
350 if (token_output) { in BuildComputation()
351 elems.push_back(*token_output); in BuildComputation()
1317 std::unique_ptr<xla::XlaOp> token_output; in CompileGraph() local
1374 token_output.reset(new xla::XlaOp(xla::AfterAll(&builder, token_inputs))); in CompileGraph()
1386 std::move(token_output), in CompileGraph()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 235 xla::XlaOp token_output = recv_from_host_tokens.empty() in Compile() local
239 ctx, ctx->compiler()->SetNodeToken(original_node_name_, token_output)); in Compile()
|