Searched refs:type_parameters (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 89 absl::flat_hash_map<string, DataType> type_parameters; in TEST_F() local
91 func, AttrSlice(&func_instantiation_attr), &type_parameters)); in TEST_F()
93 ASSERT_EQ(5, type_parameters.size()); in TEST_F()
94 EXPECT_EQ(DT_FLOAT, type_parameters["A"]); in TEST_F()
95 EXPECT_EQ(DT_INT32, type_parameters["B"]); in TEST_F()
96 EXPECT_EQ(DT_FLOAT, type_parameters["C:0"]); in TEST_F()
97 EXPECT_EQ(DT_INT32, type_parameters["C:1"]); in TEST_F()
98 EXPECT_EQ(DT_DOUBLE, type_parameters["D"]); in TEST_F()
|
| D | functions.cc | 158 absl::flat_hash_map<string, DataType>* type_parameters) { in InstantiationTypeParameters() argument
159 if (!type_parameters->empty()) { in InstantiationTypeParameters()
168 type_parameters->emplace(arg.type_attr(), dtype); in InstantiationTypeParameters()
176 type_parameters->emplace(absl::StrCat(arg.type_list_attr(), ":", index), in InstantiationTypeParameters()
|
| D | functions.h | 136 absl::flat_hash_map<string, DataType>* type_parameters);
|
| /external/rust/crates/futures-macro/src/ |
| D | select.rs | 103 let type_parameters = &variant_names; in declare_result_enum() localVariable
113 enum #result_ident<#(#type_parameters,)*> { in declare_result_enum()
115 #variants(#type_parameters), in declare_result_enum()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 166 absl::flat_hash_map<string, DataType> type_parameters; member
174 type_parameters == other.type_parameters && in operator ==()
199 + s.type_parameters.size() * 2 // in AbslHashValue()
207 absl::c_for_each(s.type_parameters, [&hashes](const TypeParam& type_param) { in AbslHashValue()
652 &sig->type_parameters)); in InitializeFunctionSpecializationSignature()
|