Home
last modified time | relevance | path

Searched defs:dimension_partition_counts (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dshape_partition.cc53 std::vector<int64_t> dimension_partition_counts(outer_dims.size()); in Run() local
92 const std::vector<int64_t>& dimension_partition_counts) { in GetTotalPartitionCount()
101 const Shape& shape, const std::vector<int64_t>& dimension_partition_counts) in ShapePartitionIterator()
H A Dir_function.cc253 const std::vector<int64_t>& dimension_partition_counts, in EmitCallToParallelForkJoin()