Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for MyBiasedDense (0.12 sec)

  1. tensorflow/compiler/mlir/tfr/resources/composite_ops.cc

    REGISTER_OP("MyAddN")
        .Input("inputs: N * T")
        .Output("sum: T")
        .Attr("N: int >= 1")
        .Attr("T: {numbertype, variant}")
        .SetIsCommutative()
        .SetIsAggregate();
    
    REGISTER_OP("MyBiasedDense")
        .Input("input: T")
        .Input("weight: T")
        .Input("bias: T")
        .Output("out: T")
        .Attr("T: {float, int8}")
        .Attr("act: {'', 'relu', 'relu6'} = ''");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 23 21:28:23 UTC 2020
    - 1.3K bytes
    - Viewed (0)
Back to top