Home
last modified time | relevance | path

Searched refs:reduce_operation (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4682 nvinfer1::ReduceOperation reduce_operation; in ConvertReduce() local
4684 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()