Searched defs:tpu_feature_counts (Results 1 – 1 of 1) sorted by relevance
166 std::vector<int> tpu_feature_counts(config.table_descriptor_size(), 0); in ComputeTpuFeatureCounts() local184 const std::vector<int> tpu_feature_counts = ComputeTpuFeatureCounts( in PopulateBatchSizeAndFeatureCounts() local