Searched refs:group_size_val (Results 1 – 2 of 2) sorted by relevance
36 Tensor group_size_val; in Run() local39 group_size_val = in Run()43 group_size_val = in Run()47 group_size_val = in Run()51 group_size_val = in Run()55 group_size_val = in Run()70 &group_size_val, col_ctx_->device, &group_size, in Run()
128 Tensor group_size_val = ca_->Scalar(group_size_); in ContinueAfterInputCopy() local145 &group_size_val, col_ctx_->device, &group_size_tensor_, in ContinueAfterInputCopy()154 group_size_tensor_ = group_size_val; in ContinueAfterInputCopy()