Searched refs:GetScalarConstNodeValue (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.h | 86 Status GetScalarConstNodeValue(const NodeDef& node, T* value) { in GetScalarConstNodeValue() function
94 Status GetScalarConstNodeValue(const NodeDef& node, int64* value);
96 Status GetScalarConstNodeValue(const NodeDef& node, bool* value);
|
| D | graph_utils_test.cc | 93 EXPECT_TRUE(GetScalarConstNodeValue<int64>(*int64_node, &result).ok()); in TEST()
102 EXPECT_TRUE(GetScalarConstNodeValue<bool>(*bool_node, &result).ok()); in TEST()
111 Status s = GetScalarConstNodeValue<int64>(*non_const, &result); in TEST()
122 Status s = GetScalarConstNodeValue<bool>(*int64_node, &result); in TEST()
140 Status s = GetScalarConstNodeValue<int64>(node, &result); in TEST()
|
| D | graph_utils.cc | 189 Status GetScalarConstNodeValue(const NodeDef& node, int64* value) { in GetScalarConstNodeValue() function
196 Status GetScalarConstNodeValue(const NodeDef& node, bool* value) { in GetScalarConstNodeValue() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | compute_batch_size_op.cc | 30 using grappler::graph_utils::GetScalarConstNodeValue;
127 auto s = GetScalarConstNodeValue(*batch_size_node, &batch_size); in GetBatchSizeFromBatchNode()
|