Home
last modified time | relevance | path

Searched refs:flat_keys (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dnest.cc96 auto flat_keys = make_safe(swig::Flatten(key, false)); in FlattenDictItems() local
98 size_t flat_keys_sz = PyList_Size(flat_keys.get()); in FlattenDictItems()
106 " elements. Key: ", PyObject_ToString(flat_keys.get()), in FlattenDictItems()
113 auto* flat_key = PyList_GetItem(flat_keys.get(), i); in FlattenDictItems()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_compile_ops.cc254 auto flat_keys = keys_tensor.flat<int64>(); in Compute() local
255 for (int64 i = 0; i < flat_keys.size(); ++i) { in Compute()
256 int64 key = flat_keys(i); in Compute()
Dxrt_compile_ops.cc270 auto flat_keys = keys_tensor.flat<int64>(); in Compute() local
271 for (int64 i = 0; i < flat_keys.size(); ++i) { in Compute()
272 int64 key = flat_keys(i); in Compute()
Dxrt_state_ops.h688 auto flat_keys = allocation_handle.flat<int64>(); in Compute() local
689 for (int64 i = 0; i < flat_keys.size(); ++i) { in Compute()
690 int64 key = flat_keys(i); in Compute()