Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc810 auto flat_t = t->scalar<int32>(); in InternalMakeShapeFromTensor() local
811 if (flat_t() != -1) { in InternalMakeShapeFromTensor()
817 flat_t()); in InternalMakeShapeFromTensor()
821 auto flat_t = t->scalar<int64_t>(); in InternalMakeShapeFromTensor() local
822 if (flat_t() != -1) { in InternalMakeShapeFromTensor()
828 flat_t()); in InternalMakeShapeFromTensor()
851 auto flat_t = t->flat<int32>(); in InternalMakeShapeFromTensor() local
852 for (int i = 0; i < flat_t.size(); ++i) { in InternalMakeShapeFromTensor()
853 const int32_t val = flat_t(i); in InternalMakeShapeFromTensor()
862 auto flat_t = t->flat<int64_t>(); in InternalMakeShapeFromTensor() local
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunique_dataset_op.cc164 auto flat_t = t.flat<tstring>(); in operator ()() local
167 hash = Hash64Combine(hash, Hash64(flat_t(i))); in operator ()()