Searched defs:trt_axis (Results 1 – 1 of 1) sorted by relevance
611 bool use_implicit_batch, int* trt_axis) { in ConvertAxis()2554 int trt_axis; in ConvertExpandDims() local2722 int trt_axis; in ConvertSqueeze() local4273 int trt_axis; in ConvertReduce() local4363 int trt_axis; in ConvertPack() local4564 int trt_axis; in ConvertSplitHelper() local4759 int trt_axis = 0; in ConvertConcat() local4960 int trt_axis = 0; in ConvertGather() local5941 int trt_axis; in ConvertArgMinMax() local