Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc54 auto parent_as_matrix = parent->flat_outer_dims<string>(); in HandleElementToSlice() local
58 parent_as_matrix(index, i) = std::move(element_flat(i)); in HandleElementToSlice()
61 parent_as_matrix.chip(index, 0) = element_flat; in HandleElementToSlice()
69 auto parent_as_matrix = parent->flat_outer_dims<Variant>(); in HandleElementToSlice() local
73 parent_as_matrix(index, i) = std::move(element_flat(i)); in HandleElementToSlice()
76 parent_as_matrix.chip(index, 0) = element_flat; in HandleElementToSlice()
96 auto parent_as_matrix = parent->flat_outer_dims<string>(); in HandleSliceToElement() local
100 element_flat(i) = std::move(parent_as_matrix(index, i)); in HandleSliceToElement()
103 element_flat = parent_as_matrix.chip(index, 0); in HandleSliceToElement()
110 auto parent_as_matrix = parent->flat_outer_dims<Variant>(); in HandleSliceToElement() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dqueue_base.cc45 auto parent_as_matrix = parent.flat_outer_dims<T>(); in HandleSliceToElement() local
46 element->flat<T>() = parent_as_matrix.chip(index, 0); in HandleSliceToElement()