Searched refs:launch_size (Results 1 – 1 of 1) sorted by relevance
380 llvm::Type* GetIndexTypeForKernel(const HloInstruction* hlo, int64 launch_size, in GetIndexTypeForKernel() argument403 if (!IsInt32(launch_size)) { in GetIndexTypeForKernel()435 int64 launch_size, in GetIndexTypeForKernelFromMlir() argument451 if (!IsInt32(launch_size)) { in GetIndexTypeForKernelFromMlir()1975 desc.get_index_type = [&](int64 launch_size) { in HandleFusion() argument1976 return GetIndexTypeForKernel(root, launch_size, &b_); in HandleFusion()2535 auto get_index_type = [&](int64 launch_size) { in EmitScatterFromMlir() argument2536 return GetIndexTypeForKernelFromMlir(scatter_op, launch_size, &b_); in EmitScatterFromMlir()