Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgather_functor.h42 SliceIndex slice_elems, in HandleCopies() argument
51 slice_elems = static_slice_elems; in HandleCopies()
54 const size_t slice_bytes = slice_elems * sizeof(T); in HandleCopies()
93 out_base + (batch_idx * indices_size + indices_idx) * slice_elems, in HandleCopies()
96 slice_elems, in HandleCopies()
108 batch_size * indices_size, slice_elems * sizeof(T), work); in HandleCopies()