Searched defs:out_flat (Results 1 – 9 of 9) sorted by relevance
167 auto out_flat = out->shaped<T, 4>( in Compute() local175 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); in Compute() local
178 auto out_flat = out->flat<Tout>(); in Compute() local294 auto out_flat = out->flat<Tout>(); in Compute() local
190 auto out_flat = out_values->flat<T>(); in Compute() local314 auto out_flat = out_values_t->flat<T>(); in Compute() local
129 out_flat; in Compute() local
55 auto out_flat = out->flat<T>(); in Compute() local
114 OutputBuffer out_flat = out->flat<T>(); in Compute() local
300 Piece out_flat = in Compute() local
760 auto out_flat = out->shaped<T, 3>({1, N, out->NumElements() / N}); in Compute() local
619 auto out_flat = ConcatInDim(&builder, out_slices, 0); in XLA_TYPED_TEST() local