Searched refs:SetTensorListOutput (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_list_ops.cc | 193 ctx->SetTensorListOutput(0, result); in Compile()
199 ctx->SetTensorListOutput(0, result); in Compile()
263 ctx->SetTensorListOutput(0, result); in Compile()
273 ctx->SetTensorListOutput(0, result); in Compile()
568 ctx->SetTensorListOutput(0, result); in Compile()
595 ctx->SetTensorListOutput(0, result); in Compile()
630 ctx->SetTensorListOutput(0, result); in Compile()
657 ctx->SetTensorListOutput(0, result); in Compile()
686 ctx->SetTensorListOutput(0, list_result); in Compile()
688 ctx->SetTensorListOutput(1, element_result); in Compile()
|
| H A D | identity_op.cc | 31 ctx->SetTensorListOutput(i, ctx->Input(i)); in Compile()
|
| H A D | arg_op.cc | 65 ctx->SetTensorListOutput(0, arg.handle()); in Compile()
|
| H A D | aggregate_ops.cc | 78 ctx->SetTensorListOutput(0, sum); in Compile()
|
| H A D | case_op.cc | 316 ctx->SetTensorListOutput(i, output_handle); in Compile()
|
| H A D | if_op.cc | 333 ctx->SetTensorListOutput(i, output_handle); in Compile()
|
| H A D | shape_op.cc | 412 ctx->SetTensorListOutput(0, result); in Compile()
|
| H A D | while_op.cc | 596 ctx->SetTensorListOutput(i, xla::GetTupleElement(while_result, i)); in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_kernel.h | 246 void SetTensorListOutput(int index, const xla::XlaOp& handle);
|
| H A D | graph_compiler.cc | 307 xla_op_context.SetTensorListOutput( in CompileFunctionalNode()
|
| H A D | xla_op_kernel.cc | 679 void XlaOpKernelContext::SetTensorListOutput(int index, in SetTensorListOutput() function in tensorflow::XlaOpKernelContext
|