Home
last modified time | relevance | path

Searched refs:partition_iterator (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition_test.cc184 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()
Dir_function.cc277 ShapePartitionIterator partition_iterator(shape, dimension_partition_counts); in EmitCallToParallelForkJoin() local
278 const int64 num_partitions = partition_iterator.GetTotalPartitionCount(); in EmitCallToParallelForkJoin()
298 partition_iterator.GetPartition(i); in EmitCallToParallelForkJoin()
/external/llvm/lib/Transforms/Scalar/
DSROA.cpp238 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-10.0/llvm/lib/Transforms/Scalar/
DSROA.cpp276 class partition_iterator;
277 iterator_range<partition_iterator> partitions();
359 friend class AllocaSlices::partition_iterator;
430 class AllocaSlices::partition_iterator class in AllocaSlices
431 : public iterator_facade_base<partition_iterator, std::forward_iterator_tag,
448 partition_iterator(AllocaSlices::iterator SI, AllocaSlices::iterator SE) in partition_iterator() function in AllocaSlices::partition_iterator
583 bool operator==(const partition_iterator &RHS) const { in operator ==()
603 partition_iterator &operator++() { in operator ++()
618 iterator_range<AllocaSlices::partition_iterator> AllocaSlices::partitions() { in partitions()
619 return make_range(partition_iterator(begin(), end()), in partitions()
[all …]
/external/llvm-project/llvm/lib/Transforms/Scalar/
DSROA.cpp265 class partition_iterator;
266 iterator_range<partition_iterator> partitions();
356 friend class AllocaSlices::partition_iterator;
427 class AllocaSlices::partition_iterator class in AllocaSlices
428 : public iterator_facade_base<partition_iterator, std::forward_iterator_tag,
445 partition_iterator(AllocaSlices::iterator SI, AllocaSlices::iterator SE) in partition_iterator() function in AllocaSlices::partition_iterator
580 bool operator==(const partition_iterator &RHS) const { in operator ==()
600 partition_iterator &operator++() { in operator ++()
615 iterator_range<AllocaSlices::partition_iterator> AllocaSlices::partitions() { in partitions()
616 return make_range(partition_iterator(begin(), end()), in partitions()
[all …]