• Pruthvi's avatar
    Pruthvi/sigmoid (#614) · 5885c09a
    Pruthvi authored
    * - Added sigmoid fusion pass
    - added mkldnn emitter code for sigmoid
    
    * - corrected sigmoid expected values
    - add layout assignment for sigmoid op
    
    * - added assert's in cpu fusion for sigmoid
    - style fix
    
    * remove debug prints
    
    * NGMX-371 #comment addressed PR comments - Added sigmoid unit test case with 3D input ii) support in cpu_emmiter for sigmoid to handle all input shapes
    
    * NGMX-371 #comment use shape_size() to calculate the 1d input size
    Unverified
    5885c09a
Graph_fprop_sigmoid.json 783 Bytes
[{"name":"Function_0","ops":[{"element_type":"float","inputs":[],"name":"Parameter_0","op":"Parameter","outputs":["Parameter_0_0"],"shape":[3,4]},{"element_type":"float","inputs":[],"name":"Constant_1","op":"Constant","outputs":["Constant_1_0"],"shape":[],"value":["1"]},{"inputs":["Parameter_0"],"name":"Negative_3","op":"Negative","outputs":["Negative_3_0"]},{"axes":[0,1],"inputs":["Constant_1"],"name":"Broadcast_2","op":"Broadcast","outputs":["Broadcast_2_0"],"shape":[3,4]},{"inputs":["Negative_3"],"name":"Exp_4","op":"Exp","outputs":["Exp_4_0"]},{"inputs":["Broadcast_2","Exp_4"],"name":"Add_5","op":"Add","outputs":["Add_5_0"]},{"inputs":["Broadcast_2","Add_5"],"name":"Divide_6","op":"Divide","outputs":["Divide_6_0"]}],"parameters":["Parameter_0"],"result":["Divide_6"]}]