Searched refs:original_dims (Results 1 – 2 of 2) sorted by relevance
38 const auto* original_dims = original_tensor.dims; in CreateNewTensorWithDifferentType() local39 TfLiteIntArray* dims = TfLiteIntArrayCreate(original_dims->size); in CreateNewTensorWithDifferentType()40 for (int i = 0; i < original_dims->size; ++i) { in CreateNewTensorWithDifferentType()41 dims->data[i] = original_dims->data[i]; in CreateNewTensorWithDifferentType()
4079 const nvinfer1::Dims original_dims = tensor->getDimensions(); in ConvertBiasAddInt8WithoutCalibration() local4083 (data_format == "NHWC" ? original_dims.nbDims - 1 : 0); in ConvertBiasAddInt8WithoutCalibration()4089 for (int i = 0; i < original_dims.nbDims; ++i) { in ConvertBiasAddInt8WithoutCalibration()4095 << DebugString(permutation, original_dims.nbDims); in ConvertBiasAddInt8WithoutCalibration()4101 if (channel_index != 0 || original_dims.nbDims != 3) { in ConvertBiasAddInt8WithoutCalibration()4117 reshape_dims.d[1] = original_dims.nbDims >= 2 ? 0 : 1; in ConvertBiasAddInt8WithoutCalibration()4118 reshape_dims.d[2] = original_dims.nbDims >= 3 ? -1 : 1; in ConvertBiasAddInt8WithoutCalibration()4143 if (channel_index != 0 || original_dims.nbDims != 3) { in ConvertBiasAddInt8WithoutCalibration()4151 nvinfer1::Dims reshape_dims = original_dims; in ConvertBiasAddInt8WithoutCalibration()4155 reshape_dims.d[channel_index] = original_dims.d[0]; in ConvertBiasAddInt8WithoutCalibration()[all …]