Searched refs:original_weights (Results 1 – 4 of 4) sorted by relevance
56 std::vector<float> original_weights; in TEST_F() local67 &original_weights)); in TEST_F()71 EXPECT_TRUE(original_weights.empty()); in TEST_F()81 &original_weights)); in TEST_F()85 EXPECT_EQ(kNumTrees, original_weights.size()); in TEST_F()86 EXPECT_EQ(original_weights, weights_); in TEST_F()103 &original_weights)); in TEST_F()122 std::vector<float> original_weights; in TEST_F() local134 &original_weights)); in TEST_F()138 EXPECT_EQ(kNumTrees, original_weights.size()); in TEST_F()[all …]
39 std::vector<float>* original_weights) { in DropOutTrees() argument44 if (original_weights == nullptr) { in DropOutTrees()62 original_weights->clear(); in DropOutTrees()102 original_weights->push_back(weights[dropped_tree]); in DropOutTrees()
46 std::vector<float>* original_weights);
237 std::vector<float> original_weights; in DoCompute() local260 weights, &dropped_trees, &original_weights)); in DoCompute()325 output_dropout_info(1, i) = original_weights[i]; in DoCompute()