Searched refs:output_nonbatch_dims (Results 1 – 1 of 1) sorted by relevance
2594 nvinfer1::Dims output_nonbatch_dims; in ConvertReshape() local2595 output_nonbatch_dims.nbDims = output_shape_dims_count - 1; in ConvertReshape()2597 output_nonbatch_dims.d[i - 1] = output_shape_dims[i]; in ConvertReshape()2603 << ", result_nonbatch_dims=" << DebugString(output_nonbatch_dims); in ConvertReshape()2612 !AreDimsStaticWithSameSize(input_nonbatch_dims, output_nonbatch_dims, in ConvertReshape()2621 DebugString(output_nonbatch_dims)); in ConvertReshape()2628 params->converter, input_tensor, output_nonbatch_dims, in ConvertReshape()