Searched refs:task_size_x (Results 1 – 1 of 1) sorted by relevance
289 const int task_size_x = DivideRoundUp(dst_[0]->Width() * dst_[0]->Batch(), in BindArguments() local295 RETURN_IF_ERROR(args->SetInt("task_size_x", task_size_x)); in BindArguments()297 const int task_size_spatial = task_size_x * task_size_y * task_size_z; in BindArguments()303 const int task_size_x = DivideRoundUp(dst_[0]->Width() * dst_[0]->Batch(), in GetGridSize() local314 return int3(task_size_x * task_size_y * task_size_z * task_size_s, 1, 1); in GetGridSize()316 return int3(task_size_x * task_size_y * task_size_z, task_size_s, 1); in GetGridSize()318 return int3(task_size_x, task_size_y * task_size_z, task_size_s); in GetGridSize()