Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc290 std::vector<std::pair<int64, int64>> dim_partitions = in EmitCallToParallelForkJoin() local
292 CHECK_EQ(num_partitioned_dims, dim_partitions.size()); in EmitCallToParallelForkJoin()
295 const std::pair<int64, int64>& dim_partition = dim_partitions[j]; in EmitCallToParallelForkJoin()