Searched refs:GetOneInt32Parameter (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 67 TF_RETURN_IF_ERROR(context.GetOneInt32Parameter("first_n", -1, &first_n)); in InsertLogging()
71 context.GetOneInt32Parameter("summarize", 1024, &summarize)); in InsertLogging()
|
| D | round_weights.cc | 37 context.GetOneInt32Parameter("num_steps", 256, &num_steps)); in RoundWeights()
|
| D | remove_nodes.cc | 41 context.GetOneInt32Parameter("max_inputs", 1, &max_inputs)); in RemoveNodes()
|
| D | quantize_weights.cc | 39 context.GetOneInt32Parameter("minimum_size", 1024, &minimum_size)); in QuantizeWeights()
|
| D | transform_utils_test.cc | 994 TF_EXPECT_OK(context.GetOneInt32Parameter("bar", 0, &value)); in TestGetOneInt32Parameter()
996 EXPECT_FALSE(context.GetOneInt32Parameter("foo", 0, &value).ok()); in TestGetOneInt32Parameter()
997 TF_EXPECT_OK(context.GetOneInt32Parameter("not_present", 10, &value)); in TestGetOneInt32Parameter()
999 EXPECT_FALSE(context.GetOneInt32Parameter("not_a_number", 0, &value).ok()); in TestGetOneInt32Parameter()
1000 EXPECT_FALSE(context.GetOneInt32Parameter("float", 0, &value).ok()); in TestGetOneInt32Parameter()
|
| D | transform_utils.h | 237 Status GetOneInt32Parameter(const string& name, int32 default_value,
|
| D | transform_utils.cc | 633 Status TransformFuncContext::GetOneInt32Parameter(const string& name, in GetOneInt32Parameter() function in tensorflow::graph_transforms::TransformFuncContext
|
| D | README.md | 1068 TF_RETURN_IF_ERROR(context.GetOneInt32Parameter("num_steps", 256, &num_steps));
|