Searched refs:first_shuffle_shape (Results 1 – 1 of 1) sorted by relevance
5804 nvinfer1::Dims first_shuffle_shape; in ConvertDepthSpaceShuffle() local5814 first_shuffle_shape = { in ConvertDepthSpaceShuffle()5831 first_shuffle_shape = {/*nbDims=*/5, in ConvertDepthSpaceShuffle()5850 first_shuffle->setReshapeDimensions(first_shuffle_shape); in ConvertDepthSpaceShuffle()