Searched defs:num_partitioned_dims (Results 1 – 2 of 2) sorted by relevance
63 int32_t num_partitions, int64_t* partitions, int32_t num_partitioned_dims, in __xla_cpu_runtime_ParallelForkJoin()
296 const int32_t num_partitioned_dims = dimension_partition_counts.size(); in EmitCallToParallelForkJoin() local