Searched refs:flat_keys (Results 1 – 4 of 4) sorted by relevance
96 auto flat_keys = make_safe(swig::Flatten(key, false)); in FlattenDictItems() local98 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()
254 auto flat_keys = keys_tensor.flat<int64>(); in Compute() local255 for (int64 i = 0; i < flat_keys.size(); ++i) { in Compute()256 int64 key = flat_keys(i); in Compute()
270 auto flat_keys = keys_tensor.flat<int64>(); in Compute() local271 for (int64 i = 0; i < flat_keys.size(); ++i) { in Compute()272 int64 key = flat_keys(i); in Compute()
688 auto flat_keys = allocation_handle.flat<int64>(); in Compute() local689 for (int64 i = 0; i < flat_keys.size(); ++i) { in Compute()690 int64 key = flat_keys(i); in Compute()