Searched refs:REDUCE_SUM (Results 1 – 7 of 7) sorted by relevance
59 if (op_type == OperationType::REDUCE_SUM || op_type == OperationType::MEAN) { in MakeOp()289 if (op_type == OperationType::REDUCE_SUM || op_type == OperationType::MEAN) { in GetReduceKernelCode()338 if (op_type == OperationType::REDUCE_SUM || in GetReduceKernelCode()391 if (op_type == OperationType::REDUCE_SUM || in GetReduceKernelCode()
71 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_SUM, in ReduceSumChannelsTest()
77 REDUCE_SUM, enumerator
157 case OperationType::REDUCE_SUM: in ToString()241 {"reduce_sum", OperationType::REDUCE_SUM}, in OperationTypeFromString()
2434 return std::make_unique<ReduceOperationParser>(OperationType::REDUCE_SUM); in NewOperationParser()
556 case OperationType::REDUCE_SUM: { in GPUOperationFromNode()
511 …%op2 = tosa.REDUCE_SUM(%logits) {axis=(%logits.rank-1)} : (tensor<%logits.type>) -> tensor<%op2.ty…2738 %op2 = tosa.REDUCE_SUM(op1) {reduce_axis=(%logits.rank - 1)} : (tensor<%op1.type>) -> tensor<%op2.t…2921 %output = lower_reduce_op<tosa.REDUCE_SUM>(%input, %output.shape, %reduction_indicies, keep_dims)