Searched refs:REDUCTION_SUM (Results 1 – 6 of 6) sorted by relevance
38 if (reduction == REDUCTION_SUM || reduction == MEAN) { in SoftMarginLossInferShape()
46 if (reduction != REDUCTION_SUM && reduction != MEAN) { in BinaryCrossEntroyInferShape()
199 REDUCTION_SUM = 0, enumerator
133 REDUCTION_SUM = 0,
52 {"sum", Reduction::REDUCTION_SUM},58 {Reduction::REDUCTION_SUM, "sum"},
100 enum Reduction : int64_t { REDUCTION_SUM = 0, MEAN = 1, NONE = 2 }; enumerator