Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dpadding_fifo_queue.cc321 auto element_t = element.tensor<T, NDIMS>(); in HandleElementToLargerSlice() local
328 slice_size[i] = element_t.dimension(i - 1); in HandleElementToLargerSlice()
330 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size); in HandleElementToLargerSlice()
/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc336 auto element_t = element.tensor<T, NDIMS>(); in HandleElementToLargerSlice() local
343 slice_size[i] = element_t.dimension(i - 1); in HandleElementToLargerSlice()
345 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size); in HandleElementToLargerSlice()