Searched refs:reduce_operation (Results 1 – 1 of 1) sorted by relevance
4682 nvinfer1::ReduceOperation reduce_operation; in ConvertReduce() local4684 reduce_operation = nvinfer1::ReduceOperation::kSUM; in ConvertReduce()4686 reduce_operation = nvinfer1::ReduceOperation::kPROD; in ConvertReduce()4688 reduce_operation = nvinfer1::ReduceOperation::kMAX; in ConvertReduce()4690 reduce_operation = nvinfer1::ReduceOperation::kMIN; in ConvertReduce()4692 reduce_operation = nvinfer1::ReduceOperation::kAVG; in ConvertReduce()4701 *tensor, reduce_operation, axes, keep_dims); in ConvertReduce()