Searched refs:first_shuffle (Results 1 – 1 of 1) sorted by relevance
5842 nvinfer1::IShuffleLayer* first_shuffle = in ConvertDepthSpaceShuffle() local5844 TFTRT_RETURN_ERROR_IF_NULLPTR(first_shuffle, node_def.name()); in ConvertDepthSpaceShuffle()5845 params->converter->SetLayerName(first_shuffle, node_def, "shuffle", in ConvertDepthSpaceShuffle()5848 first_shuffle->setFirstTranspose({2, 0, 1}); in ConvertDepthSpaceShuffle()5850 first_shuffle->setReshapeDimensions(first_shuffle_shape); in ConvertDepthSpaceShuffle()5851 first_shuffle->setSecondTranspose(transpose_perm); in ConvertDepthSpaceShuffle()5854 params->converter->network()->addShuffle(*first_shuffle->getOutput(0)); in ConvertDepthSpaceShuffle()5864 inputs.at(0).tensor(), first_shuffle->getOutput(0)); in ConvertDepthSpaceShuffle()5866 first_shuffle->getOutput(0), second_shuffle->getOutput(0)); in ConvertDepthSpaceShuffle()