Searched refs:gather_output (Results 1 – 4 of 4) sorted by relevance
33 xla::XlaOp* gather_output) { in XlaGather() argument78 *gather_output = in XlaGather()145 *gather_output = xla::Gather(input, indices, dim_numbers, slice_sizes); in XlaGather()
40 xla::XlaOp* gather_output);
15 id { node_name: "gather_output" }
3755 nvinfer1::ITensor* gather_output = layer->getOutput(0); in ConvertGather() local3756 nvinfer1::Dims trt_gather_output_dims = gather_output->getDimensions(); in ConvertGather()3776 TRT_TensorOrWeights(gather_output), trt_gather_output_dims, in ConvertGather()