Searched refs:num_ref_inputs_ (Results 1 – 1 of 1) sorted by relevance
177 OP_REQUIRES_OK(context, context->GetAttr("N", &num_ref_inputs_)); in RefSelectOp()189 OP_REQUIRES(context, index >= 0 && index < num_ref_inputs_, in Compute()191 num_ref_inputs_, ") but got ", index)); in Compute()202 int num_ref_inputs_; member in tensorflow::RefSelectOp