Searched refs:inputs_tensor (Results 1 – 1 of 1) sorted by relevance
6091 nvinfer1::ITensor* inputs_tensor = inputs.at(0).tensor(); in ConvertResize() local6093 TFTRT_RETURN_ERROR_IF_NULLPTR(inputs_tensor, params->node_def.name()); in ConvertResize()6128 const auto inputs_dims = inputs_tensor->getDimensions(); in ConvertResize()6139 inputs_tensor, {0, 3, 1, 2}, &inputs_tensor, node_def, "to_NCHW")); in ConvertResize()6145 output_dimensions.nbDims = inputs_tensor->getDimensions().nbDims; in ConvertResize()6147 output_dimensions.d[i] = inputs_tensor->getDimensions().d[i]; in ConvertResize()6154 params->converter->network()->addResize(*inputs_tensor); in ConvertResize()