Home
last modified time | relevance | path

Searched refs:input_is_ref (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc360 if (input_is_ref(start)) { in input()
402 DCHECK(!input_is_ref(index)); in input()
411 DCHECK(input_is_ref(index)); in mutable_input()
429 DCHECK(input_is_ref(index)); in replace_ref_input()
444 DCHECK(input_is_ref(input_index)); in forward_ref_input_to_ref_output()
583 DCHECK(input_is_ref(index)); in delete_ref_input()
602 if (!input_is_ref(start)) { in mutable_input()
627 if (!input_is_ref(start)) { in replace_ref_input()
Dop_kernel.h1208 bool input_is_ref(int index) const;
1528 inline bool OpKernelContext::input_is_ref(int index) const { in input_is_ref() function
1559 DCHECK(input_is_ref(index)); in input_ref_mutex()
/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.cc35 if (context->input_is_ref(0)) { in Compute()