Searched refs:dimension_partition_counts (Results 1 – 4 of 4) sorted by relevance
52 std::vector<int64> dimension_partition_counts(outer_dims.size()); in Run() local54 dimension_partition_counts[i] = in Run()62 if (GetTotalPartitionCount(dimension_partition_counts) < in Run()67 for (int64 i = 0; i < dimension_partition_counts.size(); ++i) { in Run()68 const int64 current_dim_partition_count = dimension_partition_counts[i]; in Run()70 GetTotalPartitionCount(dimension_partition_counts) / in Run()79 shape_.dimensions(outer_dims[i]) - dimension_partition_counts[i], in Run()82 dimension_partition_counts[i] += additional_partition_count; in Run()87 return dimension_partition_counts; in Run()91 const std::vector<int64>& dimension_partition_counts) { in GetTotalPartitionCount() argument[all …]
74 const std::vector<int64>& dimension_partition_counts);85 const std::vector<int64>& dimension_partition_counts);
241 const std::vector<int64>& dimension_partition_counts, llvm::IRBuilder<>* b, in EmitCallToParallelForkJoin() argument277 ShapePartitionIterator partition_iterator(shape, dimension_partition_counts); in EmitCallToParallelForkJoin()283 const int32 num_partitioned_dims = dimension_partition_counts.size(); in EmitCallToParallelForkJoin()
134 const std::vector<int64>& dimension_partition_counts, llvm::IRBuilder<>* b,