Searched refs:flattened_shape (Results 1 – 2 of 2) sorted by relevance
670 Shape flattened_shape = ShapeUtil::MakeTupleShape(flattened_shape_elems); in TryFlattenNestedTuples() local706 0, flattened_shape, in TryFlattenNestedTuples()718 0, flattened_shape, in TryFlattenNestedTuples()731 flattened_shape, in TryFlattenNestedTuples()
2032 flattened_shape = array_ops.concat([[-1], shape[2:]], axis=0)2033 t = array_ops.reshape(t, flattened_shape)