Searched refs:input_is_ref (Results 1 – 3 of 3) sorted by relevance
396 if (input_is_ref(index)) { in input()422 CHECK(!input_is_ref(index)); in input()430 CHECK(input_is_ref(index)); in mutable_input()446 CHECK(input_is_ref(index)); in replace_ref_input()460 CHECK(input_is_ref(input_index)); in forward_ref_input_to_ref_output()585 CHECK(input_is_ref(index)); in delete_ref_input()599 if (!input_is_ref(index)) { in mutable_input()618 if (!input_is_ref(index)) { in replace_ref_input()
1242 bool input_is_ref(int index) const;1603 inline bool OpKernelContext::input_is_ref(int index) const { in input_is_ref() function1618 DCHECK(input_is_ref(index)); in input_ref_mutex()
35 if (context->input_is_ref(0)) { in Compute()