Searched refs:group_size_tensor_ (Results 1 – 3 of 3) sorted by relevance
123 group_size_tensor_ = ca_->Scalar(col_ctx_->device->GetAllocator( in ContinueAfterInputCopy()127 &group_size_tensor_, in ContinueAfterInputCopy()135 group_size_tensor_ = group_size_val; in ContinueAfterInputCopy()251 col_params_->final_op.get(), &rf->chunk, &group_size_tensor_); in RunAsyncParts()
52 Tensor group_size_tensor_; variable
115 Tensor group_size_tensor_; variable