Searched refs:body_parameters (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 167 absl::flat_hash_map<string, AttrValue> body_parameters; member
180 if (body_parameters.size() != other.body_parameters.size()) return false; in operator ==()
182 for (const auto& lhs : body_parameters) { in operator ==()
183 auto it = other.body_parameters.find(lhs.first); in operator ==()
184 if (it == other.body_parameters.end()) return false; in operator ==()
200 + s.body_parameters.size() * 2 // in AbslHashValue()
215 absl::c_for_each(s.body_parameters, [&hashes](const BodyParam& body_param) { in AbslHashValue()
654 &sig->body_parameters)); in InitializeFunctionSpecializationSignature()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 194 absl::flat_hash_map<string, AttrValue>* body_parameters) { in InstantiationBodyParameters() argument
195 if (!body_parameters->empty()) { in InstantiationBodyParameters()
203 if (placeholder.empty() || body_parameters->contains(placeholder)) { in InstantiationBodyParameters()
210 body_parameters->insert({placeholder, *placeholder_value}); in InstantiationBodyParameters()
|
| D | functions_test.cc | 100 absl::flat_hash_map<string, AttrValue> body_parameters; in TEST_F() local
102 func, AttrSlice(&func_instantiation_attr), &body_parameters)); in TEST_F()
104 ASSERT_EQ(1, body_parameters.size()); in TEST_F()
105 EXPECT_EQ("key-value", body_parameters["key"].s()); in TEST_F()
|
| D | functions.h | 143 absl::flat_hash_map<string, AttrValue>* body_parameters);
|