Searched refs:features_size (Results 1 – 5 of 5) sorted by relevance
64 for (int i = 0; i < test.oblique().features_size(); ++i) { in Decide()
60 const int32 feat_size = oblique.features_size(); in CandidateGraphRunner()
81 for (int i = 0; i < test.oblique().features_size(); ++i) { in ObliqueInequalityDecisionNodeEvaluator()
558 const size_t features_size = sizeof(header_.adds_features); in ReadHeader() local559 CHECK_EQ(data->Tell(), sizeof(header_) - features_size); in ReadHeader()561 if (!data->ReadData(features_size, header_.adds_features)) { in ReadHeader()596 for (size_t i = 0; i < features_size / sizeof(uint64_t); ++i) in ReadHeader()599 for (size_t i = 0; i < features_size / sizeof(uint32_t); ++i) in ReadHeader()
2183 const int features_size = cached_features.OutputFeaturesSize(); in ModelClickContextScoreChunks() local2185 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelClickContextScoreChunks()2294 const int features_size = cached_features.OutputFeaturesSize(); in ModelBoundsSensitiveScoreChunks() local2296 TensorView<float>(all_features.data(), {batch_size, features_size}), in ModelBoundsSensitiveScoreChunks()