Searched refs:type_parameters (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 82 absl::flat_hash_map<string, DataType> type_parameters; in TEST_F() local
84 func, AttrSlice(&func_instantiation_attr), &type_parameters)); in TEST_F()
86 ASSERT_EQ(3, type_parameters.size()); in TEST_F()
87 EXPECT_EQ(DT_FLOAT, type_parameters["A"]); in TEST_F()
88 EXPECT_EQ(DT_INT32, type_parameters["B"]); in TEST_F()
89 EXPECT_EQ(DT_DOUBLE, type_parameters["C"]); in TEST_F()
|
| D | functions.cc | 425 absl::flat_hash_map<string, DataType>* type_parameters) { in InstantiationTypeParameters() argument
426 if (!type_parameters->empty()) { in InstantiationTypeParameters()
435 type_parameters->find(arg.type_attr()) == type_parameters->end()) { in InstantiationTypeParameters()
438 type_parameters->insert({arg.type_attr(), data_type}); in InstantiationTypeParameters()
|
| D | functions.h | 226 absl::flat_hash_map<string, DataType>* type_parameters);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 183 absl::flat_hash_map<string, DataType> type_parameters; member
191 type_parameters == other.type_parameters && in operator ==()
216 + s.type_parameters.size() * 2 // in AbslHashValue()
224 absl::c_for_each(s.type_parameters, [&hashes](const TypeParam& type_param) { in AbslHashValue()
742 &sig->type_parameters)); in InitializeFunctionSpecializationSignature()
|