Searched refs:reduce_operation (Results 1 – 1 of 1) sorted by relevance
3362 nvinfer1::ReduceOperation reduce_operation; in ConvertReduce() local3364 reduce_operation = nvinfer1::ReduceOperation::kSUM; in ConvertReduce()3366 reduce_operation = nvinfer1::ReduceOperation::kPROD; in ConvertReduce()3368 reduce_operation = nvinfer1::ReduceOperation::kMAX; in ConvertReduce()3370 reduce_operation = nvinfer1::ReduceOperation::kMIN; in ConvertReduce()3372 reduce_operation = nvinfer1::ReduceOperation::kAVG; in ConvertReduce()3381 *const_cast<nvinfer1::ITensor*>(tensor), reduce_operation, axes, in ConvertReduce()