Searched refs:original_dims (Results 1 – 1 of 1) sorted by relevance
2936 const nvinfer1::Dims original_dims = tensor->getDimensions(); in ConvertBiasAdd() local2940 (data_format == "NHWC" ? original_dims.nbDims - 1 : 0); in ConvertBiasAdd()2946 for (int i = 0; i < original_dims.nbDims; ++i) { in ConvertBiasAdd()2952 << DebugString(permutation, original_dims.nbDims); in ConvertBiasAdd()2958 if (channel_index != 0 || original_dims.nbDims != 3) { in ConvertBiasAdd()2972 reshape_dims.d[1] = original_dims.nbDims >= 2 ? 0 : 1; in ConvertBiasAdd()2973 reshape_dims.d[2] = original_dims.nbDims >= 3 ? -1 : 1; in ConvertBiasAdd()3000 if (channel_index != 0 || original_dims.nbDims != 3) { in ConvertBiasAdd()3006 nvinfer1::Dims reshape_dims = original_dims; in ConvertBiasAdd()3010 reshape_dims.d[channel_index] = original_dims.d[0]; in ConvertBiasAdd()[all …]