Searched refs:output_matrix_shape (Results 1 – 1 of 1) sorted by relevance
185 const TensorShape& output_matrix_shape = in PrepareOutputs() local187 OP_REQUIRES(context, output_matrix_shape.dims() <= 2, in PrepareOutputs()190 output_idx, output_matrix_shape.dims())); in PrepareOutputs()196 output_tensor_shape.AppendShape(output_matrix_shape); in PrepareOutputs()