Searched refs:group_size_val (Results 1 – 2 of 2) sorted by relevance
89 Tensor group_size_val(col_ctx_->output->dtype(), TensorShape({})); in Run() local92 group_size_val.scalar<float>()() = col_params_->group.group_size; in Run()95 group_size_val.scalar<double>()() = col_params_->group.group_size; in Run()98 group_size_val.scalar<int32>()() = col_params_->group.group_size; in Run()101 group_size_val.scalar<int64>()() = col_params_->group.group_size; in Run()113 &group_size_val, col_ctx_->device, &group_size, in Run()
121 Tensor group_size_val = ca_->Scalar(group_size_); in ContinueAfterInputCopy() local126 op_dev_ctx->CopyCPUTensorToDevice(&group_size_val, col_ctx_->device, in ContinueAfterInputCopy()135 group_size_tensor_ = group_size_val; in ContinueAfterInputCopy()