Home
last modified time | relevance | path

Searched refs:set_output_ref (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dvariable_ops.cc74 ctx->set_output_ref(0, var->mu(), var->tensor()); in Compute()
105 context->set_output_ref(0, &tmp_var->mu, &tmp_var->val); in Compute()
Dqueue_ops.cc71 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
Dconditional_accumulator_base_op.h63 ctx->set_output_ref(0, &mu_, accumulator_handle_.AccessTensor(ctx)); in Compute()
Dlookup_table_op.h97 ctx->set_output_ref(0, &mu_, table_handle_.AccessTensor(ctx)); in Compute()
Dstack.cc202 ctx->set_output_ref(0, stack->mu(), &stack->handle_); in Compute()
Dtensor_array_ops.cc124 ctx->set_output_ref(0, output_tensor_array->mu(), in Compute()
/external/tensorflow/tensorflow/core/common_runtime/
Dtestlib_ops.cc59 ctx->set_output_ref(0, &mu_, &output_); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dresource_op_kernel.h110 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
Dop_kernel.cc445 set_output_ref(output_index, (*params_->inputs)[input_index].mutex_if_ref, in forward_ref_input_to_ref_output()
821 void OpKernelContext::set_output_ref(int index, mutex* mu, in set_output_ref() function in tensorflow::OpKernelContext
830 Status OpKernelContext::set_output_ref(StringPiece name, mutex* mu, in set_output_ref() function in tensorflow::OpKernelContext
840 set_output_ref(start, mu, tensor_for_ref); in set_output_ref()
Dop_kernel.h1007 Status set_output_ref(StringPiece name, mutex* mu, Tensor* tensor_for_ref);
1182 void set_output_ref(int index, mutex* mu, Tensor* tensor_for_ref);
1686 ctx_->set_output_ref(i, mu, tensor_for_ref); in set_ref()