Searched defs:partition_count (Results 1 – 3 of 3) sorted by relevance
212 const Tensor* partitions, const Tensor* partition_count, in AllocateOutputs()240 Tensor partition_count; in ComputeAsync() local359 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