Searched refs:output_transposed_shape (Results 1 – 1 of 1) sorted by relevance
249 TensorShape output_transposed_shape = output_shape; in AllocateOutput() local250 output_transposed_shape.AppendShape({num_rows, num_cols}); in AllocateOutput()253 output_transposed_shape, in AllocateOutput()