Searched refs:SetTensorListOutput (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 191 ctx->SetTensorListOutput(0, result); in Compile()
197 ctx->SetTensorListOutput(0, result); in Compile()
259 ctx->SetTensorListOutput(0, result); in Compile()
269 ctx->SetTensorListOutput(0, result); in Compile()
562 ctx->SetTensorListOutput(0, result); in Compile()
589 ctx->SetTensorListOutput(0, result); in Compile()
624 ctx->SetTensorListOutput(0, result); in Compile()
651 ctx->SetTensorListOutput(0, result); in Compile()
680 ctx->SetTensorListOutput(0, list_result); in Compile()
682 ctx->SetTensorListOutput(1, element_result); in Compile()
|
| D | identity_op.cc | 31 ctx->SetTensorListOutput(i, ctx->Input(i)); in Compile()
|
| D | arg_op.cc | 65 ctx->SetTensorListOutput(0, arg.handle()); in Compile()
|
| D | aggregate_ops.cc | 78 ctx->SetTensorListOutput(0, sum); in Compile()
|
| D | if_op.cc | 327 ctx->SetTensorListOutput(i, output_handle); in Compile()
|
| D | case_op.cc | 329 ctx->SetTensorListOutput(i, output_handle); in Compile()
|
| D | shape_op.cc | 406 ctx->SetTensorListOutput(0, result); in Compile()
|
| D | while_op.cc | 592 ctx->SetTensorListOutput(i, xla::GetTupleElement(while_result, i)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 241 void SetTensorListOutput(int index, const xla::XlaOp& handle);
|
| D | graph_compiler.cc | 305 xla_op_context.SetTensorListOutput( in CompileFunctionalNode()
|
| D | xla_op_kernel.cc | 658 void XlaOpKernelContext::SetTensorListOutput(int index, in SetTensorListOutput() function in tensorflow::XlaOpKernelContext
|