Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc278 const int32 dim_partition_size = 2; in EmitCallToParallelForkJoin() local
281 num_partitioned_dims * dim_partition_size; in EmitCallToParallelForkJoin()
284 dim_partition_size * num_partitioned_dims * num_partitions; in EmitCallToParallelForkJoin()
296 const int32 index = partition_index + j * dim_partition_size; in EmitCallToParallelForkJoin()