Searched refs:ShapePartitionIterator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition_test.cc | 103 ShapePartitionIterator iterator(shape, {1}); in TEST_F()
109 ShapePartitionIterator iterator(shape, {2}); in TEST_F()
116 ShapePartitionIterator iterator(shape, {3}); in TEST_F()
128 ShapePartitionIterator iterator(shape, {1, 1}); in TEST_F()
135 ShapePartitionIterator iterator(shape, {2, 2}); in TEST_F()
184 ShapePartitionIterator partition_iterator(shape, dim_partition_counts); in TEST_F()
|
| D | shape_partition.cc | 99 ShapePartitionIterator::ShapePartitionIterator( in ShapePartitionIterator() function in xla::cpu::ShapePartitionIterator
129 std::vector<std::pair<int64, int64>> ShapePartitionIterator::GetPartition( in GetPartition()
155 int64 ShapePartitionIterator::GetTotalPartitionCount() const { in GetTotalPartitionCount()
|
| D | shape_partition.h | 82 class ShapePartitionIterator {
84 ShapePartitionIterator(const Shape& shape,
|
| D | ir_function.cc | 277 ShapePartitionIterator partition_iterator(shape, dimension_partition_counts); in EmitCallToParallelForkJoin()
|