Searched refs:partition_iterator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition_test.cc | 184 ShapePartitionIterator partition_iterator(shape, dim_partition_counts); in TEST_F() local
185 const int64 partition_count = partition_iterator.GetTotalPartitionCount(); in TEST_F()
187 const auto& dim_partition = partition_iterator.GetPartition(i); in TEST_F()
|
| D | ir_function.cc | 270 ShapePartitionIterator partition_iterator(shape, dimension_partition_counts); in EmitCallToParallelForkJoin() local
271 const int64 num_partitions = partition_iterator.GetTotalPartitionCount(); in EmitCallToParallelForkJoin()
291 partition_iterator.GetPartition(i); in EmitCallToParallelForkJoin()
|
| /external/llvm/lib/Transforms/Scalar/ |
| D | SROA.cpp | 238 class partition_iterator;
239 iterator_range<partition_iterator> partitions();
320 friend class AllocaSlices::partition_iterator;
391 class AllocaSlices::partition_iterator class in AllocaSlices
392 : public iterator_facade_base<partition_iterator, std::forward_iterator_tag,
409 partition_iterator(AllocaSlices::iterator SI, AllocaSlices::iterator SE) in partition_iterator() function in AllocaSlices::partition_iterator
543 bool operator==(const partition_iterator &RHS) const { in operator ==()
563 partition_iterator &operator++() { in operator ++()
578 iterator_range<AllocaSlices::partition_iterator> AllocaSlices::partitions() { in partitions()
579 return make_range(partition_iterator(begin(), end()), in partitions()
[all …]
|
| /external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/Scalar/ |
| D | SROA.cpp | 283 class partition_iterator;
284 iterator_range<partition_iterator> partitions();
366 friend class AllocaSlices::partition_iterator;
437 class AllocaSlices::partition_iterator class in AllocaSlices
438 : public iterator_facade_base<partition_iterator, std::forward_iterator_tag,
455 partition_iterator(AllocaSlices::iterator SI, AllocaSlices::iterator SE) in partition_iterator() function in AllocaSlices::partition_iterator
590 bool operator==(const partition_iterator &RHS) const { in operator ==()
610 partition_iterator &operator++() { in operator ++()
625 iterator_range<AllocaSlices::partition_iterator> AllocaSlices::partitions() { in partitions()
626 return make_range(partition_iterator(begin(), end()), in partitions()
[all …]
|