Searched refs:CountParameters (Results 1 – 3 of 3) sorted by relevance
608 int TransformFuncContext::CountParameters(const string& name) const { in CountParameters() function in tensorflow::graph_transforms::TransformFuncContext619 const int params_count = CountParameters(name); in GetOneStringParameter()636 const int params_count = CountParameters(name); in GetOneInt32Parameter()653 const int params_count = CountParameters(name); in GetOneInt64Parameter()670 const int params_count = CountParameters(name); in GetOneFloatParameter()688 const int params_count = CountParameters(name); in GetOneBoolParameter()
228 int CountParameters(const string& name) const;
970 EXPECT_EQ(2, context.CountParameters("foo")); in TestCountParameters()971 EXPECT_EQ(1, context.CountParameters("bar")); in TestCountParameters()972 EXPECT_EQ(0, context.CountParameters("not_present")); in TestCountParameters()