Searched defs:sum_op (Results 1 – 8 of 8) sorted by relevance
101 Operator* sum_op = in Run() local
257 auto sum_op = static_cast<const TensorFlowSumOperator*>(unary_op); in Run() local
53 auto sum_op = std::make_unique<TensorFlowSumOperator>(); in RunResolveSum() local
49 auto sum_op = new TensorFlowSumOperator; in RunIdentifyL2Normalization() local
94 auto sum_op = mlir::cast<mlir::TF::UnsortedSegmentSumOp>(op); in ExpandOp() local
239 dnnl::sum sum_op(sum_pd); in Compute() local
393 auto sum_op = ops::Sum(scope.WithOpName("sum").WithDevice("/device:GPU:0"), in ReduceTransposerKeepDims() local460 auto sum_op = ops::Max(scope.WithOpName("max").WithDevice("/device:GPU:0"), in ReduceTransposerValidAxisNode() local
928 tensorflow::NodeDef* sum_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local