Searched refs:num_partitioned_dims (Results 1 – 3 of 3) sorted by relevance
65 int64* partitions, int32 num_partitioned_dims, void* function_ptr) { in __xla_cpu_runtime_ParallelForkJoin() argument68 << " num_partitioned_dims: " << num_partitioned_dims; in __xla_cpu_runtime_ParallelForkJoin()71 CHECK_GT(num_partitioned_dims, 0); in __xla_cpu_runtime_ParallelForkJoin()82 const int64 stride = 2 * num_partitioned_dims; in __xla_cpu_runtime_ParallelForkJoin()
276 const int32 num_partitioned_dims = dimension_partition_counts.size(); in EmitCallToParallelForkJoin() local281 num_partitioned_dims * dim_partition_size; in EmitCallToParallelForkJoin()284 dim_partition_size * num_partitioned_dims * num_partitions; in EmitCallToParallelForkJoin()292 CHECK_EQ(num_partitioned_dims, dim_partitions.size()); in EmitCallToParallelForkJoin()294 for (int32 j = 0; j < num_partitioned_dims; ++j) { in EmitCallToParallelForkJoin()323 fork_join_arguments.push_back(b->getInt32(num_partitioned_dims)); in EmitCallToParallelForkJoin()
29 tensorflow::int32 num_partitioned_dims, void* function_ptr);