Searched refs:first_row_shape_ (Results 1 – 2 of 2) sorted by relevance
213 std::vector<TensorShape> first_row_shape_; variable
609 *shapes = first_row_shape_; in GetOutputShapes()686 std::transform(first_row.begin(), first_row.end(), std::back_inserter(first_row_shape_), in GetFirstRowShapeAndType()