Searched refs:flat_t (Results 1 – 3 of 3) sorted by relevance
846 auto flat_t = t->scalar<int32>(); in InternalMakeShapeFromTensor() local847 if (flat_t() != -1) { in InternalMakeShapeFromTensor()853 flat_t()); in InternalMakeShapeFromTensor()857 auto flat_t = t->scalar<int64>(); in InternalMakeShapeFromTensor() local858 if (flat_t() != -1) { in InternalMakeShapeFromTensor()864 flat_t()); in InternalMakeShapeFromTensor()887 auto flat_t = t->flat<int32>(); in InternalMakeShapeFromTensor() local888 for (int i = 0; i < flat_t.size(); ++i) { in InternalMakeShapeFromTensor()889 const int32 val = flat_t(i); in InternalMakeShapeFromTensor()898 auto flat_t = t->flat<int64>(); in InternalMakeShapeFromTensor() local[all …]
174 auto flat_t = t.flat<string>(); in operator ()() local177 hash = Hash64Combine(hash, Hash64(flat_t(i))); in operator ()()
124 flat_t = nest.flatten(t)125 with ops.name_scope(name, "tile_batch", flat_t + [multiplier]):