Searched refs:CrossentropyCategorical1Hot (Results 1 – 5 of 5) sorted by relevance
3 CrossentropyCategorical1Hot, CrossentropyCategorical1HotGrad,
1414 class CrossentropyCategorical1Hot(gof.Op): class1475 crossentropy_categorical_1hot = CrossentropyCategorical1Hot()
24 CrossentropyCategorical1Hot,565 [CrossentropyCategorical1Hot()(admat, alvec)],567 CrossentropyCategorical1Hot)
1452 @op_lifter([tensor.nnet.CrossentropyCategorical1Hot])1453 @register_opt2([tensor.nnet.CrossentropyCategorical1Hot], 'fast_compile')
768 assert not any(isinstance(n.op, (tensor.nnet.CrossentropyCategorical1Hot,