Searched refs:set_output_ref (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | variable_ops.cc | 74 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()
|
D | queue_ops.cc | 71 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
|
D | conditional_accumulator_base_op.h | 63 ctx->set_output_ref(0, &mu_, accumulator_handle_.AccessTensor(ctx)); in Compute()
|
D | lookup_table_op.h | 97 ctx->set_output_ref(0, &mu_, table_handle_.AccessTensor(ctx)); in Compute()
|
D | stack.cc | 202 ctx->set_output_ref(0, stack->mu(), &stack->handle_); in Compute()
|
D | tensor_array_ops.cc | 124 ctx->set_output_ref(0, output_tensor_array->mu(), in Compute()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | testlib_ops.cc | 59 ctx->set_output_ref(0, &mu_, &output_); in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | resource_op_kernel.h | 110 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
|
D | op_kernel.cc | 445 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()
|
D | op_kernel.h | 1007 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()
|