Searched refs:specified_output_shape (Results 1 – 1 of 1) sorted by relevance
239 std::vector<int32> const& specified_output_shape = in ProcessTransposeConvOperator() local242 *(output_array.mutable_shape()->mutable_dims()) = specified_output_shape; in ProcessTransposeConvOperator()249 Shape{specified_output_shape[0], specified_output_shape[1], in ProcessTransposeConvOperator()250 specified_output_shape[2], input_depth * kheight * kwidth}); in ProcessTransposeConvOperator()