Searched refs:allow_small_batch_ (Results 1 – 1 of 1) sorted by relevance
577 context->GetAttr("allow_small_batch", &allow_small_batch_)); in TakeManyOp()601 num_elements, allow_small_batch_, timeout_, ctx, in ComputeAsync()625 bool allow_small_batch_; member in tensorflow::barrier::TakeManyOp