HomeSort by relevance Sort by last modified time
    Searched refs:dim_partition (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
shape_partition_test.cc 227 const auto& dim_partition = partition_iterator.GetPartition(i); local
228 for (int dim = 0; dim < dim_partition.size(); ++dim) {
230 std::make_pair(dim_partition[dim].first,
231 dim_partition[dim].first + dim_partition[dim].second));
ir_function.cc 293 const std::pair<int64, int64>& dim_partition = dim_partitions[j]; local
296 partitions[index] = ir_builder->getInt64(dim_partition.first);
298 ir_builder->getInt64(dim_partition.first + dim_partition.second);

Completed in 189 milliseconds