Home
last modified time | relevance | path

Searched refs:CountParameters (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.cc608 int TransformFuncContext::CountParameters(const string& name) const { in CountParameters() function in tensorflow::graph_transforms::TransformFuncContext
619 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()
Dtransform_utils.h228 int CountParameters(const string& name) const;
Dtransform_utils_test.cc970 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()