Searched refs:shuffle_layer (Results 1 – 1 of 1) sorted by relevance
2959 nvinfer1::IShuffleLayer* shuffle_layer = in ConvertBiasAdd() local2961 TFTRT_RETURN_ERROR_IF_NULLPTR(shuffle_layer, node_def.name()); in ConvertBiasAdd()2963 tensor, shuffle_layer->getOutput(0)); in ConvertBiasAdd()2974 shuffle_layer->setReshapeDimensions(reshape_dims); in ConvertBiasAdd()2977 shuffle_layer->setFirstTranspose(permutation); in ConvertBiasAdd()2979 tensor = shuffle_layer->getOutput(0); in ConvertBiasAdd()3001 nvinfer1::IShuffleLayer* shuffle_layer = in ConvertBiasAdd() local3003 TFTRT_RETURN_ERROR_IF_NULLPTR(shuffle_layer, node_def.name()); in ConvertBiasAdd()3013 shuffle_layer->setReshapeDimensions(reshape_dims); in ConvertBiasAdd()3016 shuffle_layer->setSecondTranspose(permutation); in ConvertBiasAdd()[all …]