Searched refs:FindConfigValueForKey (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_flag_utils_test.cc | 32 using tensorflow::tensor_flag_utils::FindConfigValueForKey;
235 TEST(SparseUtils, FindConfigValueForKey) { in TEST() argument
239 auto val = FindConfigValueForKey<float, int32>(config_mat, {70, 40}); in TEST()
241 val = FindConfigValueForKey<float, int32>(config_mat, {60, 50}); in TEST()
243 val = FindConfigValueForKey<float, int32>(config_mat, {60, 60}); in TEST()
245 val = FindConfigValueForKey<float, int32>(config_mat, {60, 40}); in TEST()
247 val = FindConfigValueForKey<float, int32>(config_mat, {50, 60}); in TEST()
249 val = FindConfigValueForKey<float, int32>(config_mat, {20, 30}); in TEST()
251 val = FindConfigValueForKey<float, int32>(config_mat, {30, 10}); in TEST()
257 auto val = FindConfigValueForKey<float, int64>(config_mat, {70, 40}); in TEST()
[all …]
|
| D | tensor_flag_utils.cc | 76 MatrixType FindConfigValueForKey( in FindConfigValueForKey() function
133 MatrixType FindConfigValueForKey( in FindConfigValueForKey() function
163 template float FindConfigValueForKey<float, TypeIndex>( \
166 template float FindConfigValueForKey<float, TypeIndex>( \
168 template int64 FindConfigValueForKey<int64, TypeIndex>( \
|
| D | tensor_flag_utils.h | 53 MatrixType FindConfigValueForKey(
62 MatrixType FindConfigValueForKey(
|