Searched refs:launch_order (Results 1 – 2 of 2) sorted by relevance
153 std::vector<HloInstruction*>* launch_order) { in BFSLaunchOrder() argument177 launch_order->push_back(x); in BFSLaunchOrder()
60 int3 launch_order; member70 launch_remap[params.launch_order.x] = 0; in GlobalIdsGen()71 launch_remap[params.launch_order.y] = 1; in GlobalIdsGen()72 launch_remap[params.launch_order.z] = 2; in GlobalIdsGen()83 if (params.launch_order.x == 0) { in GlobalIdsGen()93 if (params.launch_order.y == 1) { in GlobalIdsGen()102 if (params.launch_order.x == 0) { in GlobalIdsGen()110 if (params.launch_order.y == 1) { in GlobalIdsGen()118 if (params.launch_order.z == 2) { in GlobalIdsGen()171 ids_params.launch_order = params.work_group_launch_order; in GenerateConvolution()