Searched refs:body_parameters (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 453 absl::flat_hash_map<string, AttrValue>* body_parameters) { in InstantiationBodyParameters() argument
454 if (!body_parameters->empty()) { in InstantiationBodyParameters()
463 body_parameters->find(placeholder) != body_parameters->end()) { in InstantiationBodyParameters()
470 body_parameters->insert({placeholder, *placeholder_value}); in InstantiationBodyParameters()
|
| D | functions_test.cc | 91 absl::flat_hash_map<string, AttrValue> body_parameters; in TEST_F() local
93 func, AttrSlice(&func_instantiation_attr), &body_parameters)); in TEST_F()
95 ASSERT_EQ(1, body_parameters.size()); in TEST_F()
96 EXPECT_EQ("key-value", body_parameters["key"].s()); in TEST_F()
|
| D | functions.h | 233 absl::flat_hash_map<string, AttrValue>* body_parameters);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 184 absl::flat_hash_map<string, AttrValue> body_parameters; member
197 if (body_parameters.size() != other.body_parameters.size()) return false; in operator ==()
199 for (const auto& lhs : body_parameters) { in operator ==()
200 auto it = other.body_parameters.find(lhs.first); in operator ==()
201 if (it == other.body_parameters.end()) return false; in operator ==()
217 + s.body_parameters.size() * 2 // in AbslHashValue()
232 absl::c_for_each(s.body_parameters, [&hashes](const BodyParam& body_param) { in AbslHashValue()
744 &sig->body_parameters)); in InitializeFunctionSpecializationSignature()
|