Searched defs:trt_axis (Results 1 – 1 of 1) sorted by relevance
551 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()2306 int trt_axis; in ConvertExpandDims() local2489 int trt_axis; in ConvertSqueeze() local3607 int trt_axis; in ConvertReduce() local3697 int trt_axis{0}; in ConvertPack() local3904 int trt_axis; in ConvertSplitHelper() local4117 int trt_axis = 0; in ConvertConcat() local4349 int trt_axis = 0; in ConvertGather() local4799 int trt_axis; in ConvertArgMinMax() local