Searched defs:trt_axis (Results 1 – 1 of 1) sorted by relevance
530 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()2656 int trt_axis; in ConvertExpandDims() local2819 int trt_axis; in ConvertSqueeze() local4675 int trt_axis; in ConvertReduce() local4769 int trt_axis; in ConvertPack() local4967 int trt_axis; in ConvertSplitHelper() local5134 int trt_axis = 0; in ConvertConcat() local5333 int trt_axis = 0; in ConvertGather() local5684 int trt_axis; in ConvertArgMinMax() local