Searched refs:selected_input (Results 1 – 2 of 2) sorted by relevance
172 int64 selected_input = selector_result[0].scalar<int64>()(); in GetNextInternal() local173 if (selected_input < 0 || selected_input >= data_input_impls_.size()) { in GetNextInternal()175 "Selector index out of range: ", selected_input, in GetNextInternal()179 if (data_input_impls_[selected_input]) { in GetNextInternal()181 TF_RETURN_IF_ERROR(data_input_impls_[selected_input]->GetNext( in GetNextInternal()188 data_input_impls_[selected_input].reset(); in GetNextInternal()199 << selected_input; in GetNextInternal()
949 Value selected_input = builder->create<SelectOp>( in BuildArgMinMaxReductionBody() local954 Value return_values[] = {selected_input, selected_index}; in BuildArgMinMaxReductionBody()