Searched defs:launch_size (Results 1 – 1 of 1) sorted by relevance
380 llvm::Type* GetIndexTypeForKernel(const HloInstruction* hlo, int64 launch_size, in GetIndexTypeForKernel()435 int64 launch_size, in GetIndexTypeForKernelFromMlir()1975 desc.get_index_type = [&](int64 launch_size) { in HandleFusion()2535 auto get_index_type = [&](int64 launch_size) { in EmitScatterFromMlir()