Searched defs:partition_count (Results 1 – 11 of 11) sorted by relevance
66 int partition_count; member222 int partition_count; // 1 to 4; Zero marks a constant-color block. member276 int partition_count in get_partition_table()
127 int partition_count = read_bits(2, 11, pb.data) + 1; in physical_to_symbolic() local
202 int partition_count = scb->partition_count; in decompress_symbolic_block() local
227 int partition_count, in generate_one_partition_table()
58 int64 partition_count = ag_size / shard_size; in DecomposeAllGather() local
46 int64 partition_count) { in ReplaceReplicatedAllReduce()
219 const Tensor* partitions, const Tensor* partition_count, in AllocateOutputs()247 Tensor partition_count; in ComputeAsync() local369 Tensor* partition_count, Tensor* indices_out, in CountAndSortParts()
56 gtl::InlinedVector<int, 32> partition_count(num_partitions_); in ValidateAndAllocateOutputs() local
185 const int64 partition_count = partition_iterator.GetTotalPartitionCount(); in TEST_F() local
668 auto partition_count = in PartitionBaseCase() local1063 auto partition_count = in PartitionBaseCase() local
1662 const int64 partition_count = in HandleSort() local