Searched refs:second_shuffle (Results 1 – 1 of 1) sorted by relevance
5853 nvinfer1::IShuffleLayer* second_shuffle = in ConvertDepthSpaceShuffle() local5855 TFTRT_RETURN_ERROR_IF_NULLPTR(second_shuffle, node_def.name()); in ConvertDepthSpaceShuffle()5856 params->converter->SetLayerName(second_shuffle, node_def, "shuffle", in ConvertDepthSpaceShuffle()5858 second_shuffle->setReshapeDimensions(second_shuffle_shape); in ConvertDepthSpaceShuffle()5860 second_shuffle->setSecondTranspose({1, 2, 0}); in ConvertDepthSpaceShuffle()5866 first_shuffle->getOutput(0), second_shuffle->getOutput(0)); in ConvertDepthSpaceShuffle()5867 params->outputs->push_back(TRT_TensorOrWeights(second_shuffle->getOutput(0))); in ConvertDepthSpaceShuffle()