Searched defs:reduction_enum (Results 1 – 1 of 1) sorted by relevance
51 torch::Reduction::Reduction reduction_enum; variable156 auto reduction_enum = enumtype::reduction_get_enum(reduction); in binary_cross_entropy() local