Home
last modified time | relevance | path

Searched defs:keys_flat (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/lookup_tables/
Dgeneric_table_op_kernels.cc71 const auto keys_flat = keys.flat<Variant>(); in TensorInsertHelper() local
123 const auto keys_flat = keys.flat<KeyType>(); in TensorLookupHelper() local
140 const auto keys_flat = keys.flat<Variant>(); in TensorLookupHelper() local
Dflat_hash_map_op_kernels.cc210 auto keys_flat = keys.flat<Variant>(); in AllocateContainer() local