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
168 auto out_flat = out->flat<Tout>(); in Compute() local279 auto out_flat = out->flat<Tout>(); in Compute() local
185 auto out_flat = out_values->flat<T>(); in Compute() local293 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
236 Piece out_flat = in Compute() local
114 OutputBuffer out_flat = out->flat<T>(); in Compute() local
695 auto out_flat = out->shaped<T, 3>({1, N, out->NumElements() / N}); in Compute() local
618 auto out_flat = ConcatInDim(&builder, out_slices, 0); in XLA_TYPED_TEST() local