Searched defs:sum_op (Results 1 – 13 of 13) sorted by relevance
51 struct sum_op struct54 int operator()(T const & elem, int value) const in operator ()()60 int operator()(T & elem, int value) const in operator ()()66 typedef int result_type;
61 SumOp<float> sum_op; in reduce_sum() local118 SumOp<float> sum_op; in max_scale_exp2_sum() local168 SumOp<float> sum_op; member
44 pytorch_flash::SumOp<float> sum_op; in dot_do_o() local
1153 SumOp<float> sum_op; in combine_attn_seqk_parallel() local
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
506 CustomSum sum_op{}; in _segment_reduce_lengths_offsets_cuda_kernel() local542 CustomSum sum_op{}; in _segment_reduce_lengths_offsets_cuda_kernel() 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