Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc55 auto element_flat = element.flat<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()
70 auto element_flat = element.flat<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()
97 auto element_flat = element->flat<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()
111 auto element_flat = element->flat<Variant>(); in HandleSliceToElement() local
[all …]