Searched refs:SetTensorListOutput (Results 1 – 4 of 4) sorted by relevance
116 ctx->SetTensorListOutput(0, output_list); in Compile()152 ctx->SetTensorListOutput(0, output_list); in Compile()298 ctx->SetTensorListOutput(0, output_list); in Compile()374 ctx->SetTensorListOutput(0, output_list); in Compile()418 ctx->SetTensorListOutput(0, output_list); in Compile()462 ctx->SetTensorListOutput(0, output_list); in Compile()
57 ctx->SetTensorListOutput(0, arg.handle()); in Compile()
181 void SetTensorListOutput(int index, const xla::XlaOp& handle);
491 void XlaOpKernelContext::SetTensorListOutput(int index, in SetTensorListOutput() function in tensorflow::XlaOpKernelContext