Searched refs:second_shuffle_shape_tensor (Results 1 – 1 of 1) sorted by relevance
5103 ITensorProxyPtr second_shuffle_shape_tensor; in ConvertDepthSpaceShuffle() local5132 second_shuffle_shape_tensor = result.ValueOrDie().second; in ConvertDepthSpaceShuffle()5167 second_shuffle->setInput(1, *second_shuffle_shape_tensor->trt_tensor()); in ConvertDepthSpaceShuffle()