Searched refs:CountParameters (Results 1 – 3 of 3) sorted by relevance
616 int TransformFuncContext::CountParameters(const string& name) const { in CountParameters() function in tensorflow::graph_transforms::TransformFuncContext627 const int params_count = CountParameters(name); in GetOneStringParameter()644 const int params_count = CountParameters(name); in GetOneInt32Parameter()661 const int params_count = CountParameters(name); in GetOneInt64Parameter()678 const int params_count = CountParameters(name); in GetOneFloatParameter()696 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()