Searched refs:shuffle_layer (Results 1 – 1 of 1) sorted by relevance
4102 nvinfer1::IShuffleLayer* shuffle_layer = in ConvertBiasAddInt8WithoutCalibration() local4104 TFTRT_RETURN_ERROR_IF_NULLPTR(shuffle_layer, node_def.name()); in ConvertBiasAddInt8WithoutCalibration()4105 params->converter->SetLayerName(shuffle_layer, node_def, "shuffle", in ConvertBiasAddInt8WithoutCalibration()4108 tensor, shuffle_layer->getOutput(0)); in ConvertBiasAddInt8WithoutCalibration()4119 shuffle_layer->setReshapeDimensions(reshape_dims); in ConvertBiasAddInt8WithoutCalibration()4122 shuffle_layer->setFirstTranspose(permutation); in ConvertBiasAddInt8WithoutCalibration()4124 tensor = shuffle_layer->getOutput(0); in ConvertBiasAddInt8WithoutCalibration()4144 nvinfer1::IShuffleLayer* shuffle_layer = in ConvertBiasAddInt8WithoutCalibration() local4146 TFTRT_RETURN_ERROR_IF_NULLPTR(shuffle_layer, node_def.name()); in ConvertBiasAddInt8WithoutCalibration()4147 params->converter->SetLayerName(shuffle_layer, node_def, "shuffle", in ConvertBiasAddInt8WithoutCalibration()[all …]