Searched refs:tiled_param_ids (Results 1 – 2 of 2) sorted by relevance
245 absl::Span<const int64> tiled_param_ids);
3051 HloInstruction* unnested_hlo, absl::Span<const int64> tiled_param_ids, in EmitKernel() argument3060 for (int64 id : tiled_param_ids) { in EmitKernel()3146 if (!tiled_param_ids.empty()) { in EmitKernel()3156 for (int64 id : tiled_param_ids) { in EmitKernel()3190 if (block_contains_multi_tiles && !tiled_param_ids.empty()) { in EmitKernel()3238 absl::Span<const int64> tiled_param_ids) { in EmitHlo021Tile() argument3266 return EmitKernel(hlo, tiled_param_ids, kernel_generator, &kernel_info); in EmitHlo021Tile()