Home
last modified time | relevance | path

Searched refs:FindConfigValueForKey (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtensor_flag_utils_test.cc32 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 …]
Dtensor_flag_utils.cc76 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>( \
Dtensor_flag_utils.h53 MatrixType FindConfigValueForKey(
62 MatrixType FindConfigValueForKey(