Searched defs:num_partitioned_dims (Results 1 – 2 of 2) sorted by relevance
65 int64* partitions, int32 num_partitioned_dims, void* function_ptr) { in __xla_cpu_runtime_ParallelForkJoin()
276 const int32 num_partitioned_dims = dimension_partition_counts.size(); in EmitCallToParallelForkJoin() local