Searched refs:trt_axis (Results 1 – 1 of 1) sorted by relevance
505 int* trt_axis) { in ConvertAxis() argument522 *trt_axis = tf_axis - 1; in ConvertAxis()3725 int trt_axis = 0; in ConvertGather() local3727 node_def.name(), &trt_axis)); in ConvertGather()3752 *const_cast<nvinfer1::ITensor*>(indices_tensor.tensor()), trt_axis); in ConvertGather()3768 for (int i = trt_gather_output_dims.nbDims; i > trt_axis; --i) { in ConvertGather()3771 trt_gather_output_dims.d[trt_axis] = 1; in ConvertGather()