Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition.cc26 int64 outer_dim_size = 1; in Run() local
32 outer_dim_size *= shape_.dimensions(dimension); in Run()
33 if (outer_dim_size >= target_partition_count) { in Run()
39 target_partition_count = std::min(outer_dim_size, target_partition_count); in Run()