Searched defs:input_feature_batch_sizes (Results 1 – 1 of 1) sorted by relevance
135 std::vector<int32> input_feature_batch_sizes; in ComputeInputFeatureBatchSizes() local148 absl::Span<const int> input_feature_batch_sizes) { in ComputeBatchSizePerTensorCore()163 absl::Span<const int> input_feature_batch_sizes, in ComputeTpuFeatureCounts()180 const std::vector<int> input_feature_batch_sizes = in PopulateBatchSizeAndFeatureCounts() local