Searched refs:BoostedTreesEnsembleResource (Results 1 – 7 of 7) sorted by relevance
27 BoostedTreesEnsembleResource::BoostedTreesEnsembleResource() in BoostedTreesEnsembleResource() function in tensorflow::BoostedTreesEnsembleResource32 string BoostedTreesEnsembleResource::DebugString() const { in DebugString()37 bool BoostedTreesEnsembleResource::InitFromSerialized(const string& serialized, in InitFromSerialized()47 string BoostedTreesEnsembleResource::SerializeAsString() const { in SerializeAsString()51 int32 BoostedTreesEnsembleResource::num_trees() const { in num_trees()55 int32 BoostedTreesEnsembleResource::next_node( in next_node()85 std::vector<float> BoostedTreesEnsembleResource::node_value( in node_value()121 void BoostedTreesEnsembleResource::set_node_value(const int32 tree_id, in set_node_value()131 int32 BoostedTreesEnsembleResource::GetNumLayersGrown( in GetNumLayersGrown()137 void BoostedTreesEnsembleResource::SetNumLayersGrown( in SetNumLayersGrown()[all …]
29 REGISTER_RESOURCE_HANDLE_KERNEL(BoostedTreesEnsembleResource);33 IsResourceInitialized<BoostedTreesEnsembleResource>);51 std::unique_ptr<BoostedTreesEnsembleResource> result( in Compute()52 new BoostedTreesEnsembleResource()); in Compute()82 core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource; in Compute()144 core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource; in Compute()171 core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource; in Compute()
47 core::RefCountPtr<BoostedTreesEnsembleResource> ensemble_resource; in Compute()150 const core::RefCountPtr<BoostedTreesEnsembleResource>& resource) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()243 core::RefCountPtr<BoostedTreesEnsembleResource> ensemble_resource; in Compute()370 const core::RefCountPtr<BoostedTreesEnsembleResource>& resource) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()464 core::RefCountPtr<BoostedTreesEnsembleResource> ensemble_resource; in Compute()
49 class BoostedTreesEnsembleResource : public StampedResource {51 BoostedTreesEnsembleResource();
68 core::RefCountPtr<BoostedTreesEnsembleResource> resource; in Compute()211 core::RefCountPtr<BoostedTreesEnsembleResource> resource; in Compute()311 core::RefCountPtr<BoostedTreesEnsembleResource> resource; in Compute()
4 summary: "Creates a handle to a BoostedTreesEnsembleResource"
31 REGISTER_RESOURCE_HANDLE_OP(BoostedTreesEnsembleResource);