Searched defs:dimension_partition_counts (Results 1 – 2 of 2) sorted by relevance
52 std::vector<int64> dimension_partition_counts(outer_dims.size()); in Run() local91 const std::vector<int64>& dimension_partition_counts) { in GetTotalPartitionCount()100 const Shape& shape, const std::vector<int64>& dimension_partition_counts) in ShapePartitionIterator()
241 const std::vector<int64>& dimension_partition_counts, llvm::IRBuilder<>* b, in EmitCallToParallelForkJoin()