Searched defs:launch_size (Results 1 – 1 of 1) sorted by relevance
326 int64_t launch_size, llvm::IRBuilder<>* b) { in GetIndexTypeForKernel()379 llvm::Type* GetIndexTypeForKernel(mlir::Operation* op, int64_t launch_size, in GetIndexTypeForKernel()2249 auto get_index_type = [&](int64_t launch_size) { in EmitScatter()5285 desc.get_index_type = [&](int64_t launch_size) { in EmitScatter()