Searched refs:features_size (Results 1 – 1 of 1) sorted by relevance
3002 const int features_size = cached_features.OutputFeaturesSize(); in ModelClickContextScoreChunks() local3004 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelClickContextScoreChunks()3111 const int features_size = cached_features.OutputFeaturesSize(); in ModelBoundsSensitiveScoreChunks() local3113 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelBoundsSensitiveScoreChunks()