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