Searched defs:gather_output (Results 1 – 2 of 2) sorted by relevance
33 xla::XlaOp* gather_output) { in XlaGather()
3755 nvinfer1::ITensor* gather_output = layer->getOutput(0); in ConvertGather() local