Home
last modified time | relevance | path

Searched refs:SetResourceOutput (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/aot/
Daot_only_var_handle_op.cc45 context->SetResourceOutput(0, resource.get()); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h193 void SetResourceOutput(int index, XlaResource* resource);
Dxla_op_kernel.cc496 void XlaOpKernelContext::SetResourceOutput(int index, XlaResource* resource) { in SetResourceOutput() function in tensorflow::XlaOpKernelContext
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc113 ctx->SetResourceOutput(0, resource); in Compile()
Dtensor_array_ops.cc173 ctx->SetResourceOutput(0, var); in Compile()
594 ctx->SetResourceOutput(0, gradient); in Compile()