Searched defs:out_flat (Results 1 – 8 of 8) sorted by relevance
107 auto out_flat = out->flat<Tout>(); in Compute() local218 auto out_flat = out->flat<Tout>(); in Compute() local
108 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); 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
114 OutputBuffer out_flat = out->flat<T>(); in Compute() local
691 auto out_flat = out->shaped<T, 3>({1, N, out->NumElements() / N}); in Compute() local
574 auto out_flat = ConcatInDim(&builder, out_slices, 0); in XLA_TYPED_TEST() local