CASE {"name":"constant/float16/float16/rank1/bias_identity","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank1/unit_weight_control","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank1/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank1/bias_jvp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank1/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank2/bias_identity","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank2/unit_weight_control","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank2/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank2/bias_jvp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank2/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank3/bias_identity","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank3/unit_weight_control","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank3/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank3/bias_jvp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float16/rank3/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float16/bfloat16/rank1/bias_identity","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float16/bfloat16/rank1/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank1/bias_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank1/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float16/bfloat16/rank2/bias_identity","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float16/bfloat16/rank2/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank2/bias_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank2/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float16/bfloat16/rank3/bias_identity","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float16/bfloat16/rank3/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank3/bias_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity","Infinity",1.0078125,0,"-Infinity"],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":false} CASE {"name":"constant/float16/bfloat16/rank3/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float32/rank1/bias_identity","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float32/rank1/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank1/bias_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank1/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float32/rank2/bias_identity","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float32/rank2/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank2/bias_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank2/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float32/rank3/bias_identity","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float32/rank3/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank3/bias_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float32/rank3/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float64/rank1/bias_identity","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float64/rank1/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank1/bias_jvp","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank1/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float64/rank2/bias_identity","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float64/rank2/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank2/bias_jvp","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank2/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float64/rank3/bias_identity","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float16/float64/rank3/unit_weight_invariance","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank3/bias_jvp","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity","Infinity",1,0,"-Infinity"],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float16/float64/rank3/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float16/rank1/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float16/rank1/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank1/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank1/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float16/rank2/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float16/rank2/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank2/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank2/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float16/rank3/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float16/rank3/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank3/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float16/rank3/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank1/bias_identity","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank1/unit_weight_control","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank1/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank1/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank1/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank2/bias_identity","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank2/unit_weight_control","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank2/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank2/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank2/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank3/bias_identity","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank3/unit_weight_control","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank3/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank3/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/bfloat16/rank3/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float32/rank1/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float32/rank1/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank1/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank1/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float32/rank2/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float32/rank2/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank2/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank2/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float32/rank3/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float32/rank3/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank3/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144,70144,1,9.3132257461547852e-10,-70144],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float32/rank3/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float64/rank1/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float64/rank1/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank1/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank1/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float64/rank2/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float64/rank2/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank2/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank2/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float64/rank3/bias_identity","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/bfloat16/float64/rank3/unit_weight_invariance","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank3/bias_jvp","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/bfloat16/float64/rank3/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank1/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank1/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank1/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank1/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank2/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank2/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank2/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank2/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank3/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank3/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank3/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float16/rank3/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank1/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank1/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank1/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank1/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank2/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank2/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank2/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank2/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank3/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank3/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank3/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float32/bfloat16/rank3/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank1/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank1/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank1/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank1/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank1/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank2/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank2/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank2/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank2/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank2/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank3/bias_identity","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank3/unit_weight_control","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank3/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank3/bias_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float32/rank3/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float64/rank1/bias_identity","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float64/rank1/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank1/bias_jvp","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank1/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float64/rank2/bias_identity","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float64/rank2/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank2/bias_jvp","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank2/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float64/rank3/bias_identity","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float32/float64/rank3/unit_weight_invariance","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank3/bias_jvp","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624,1125899906842624,1,0,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":false} CASE {"name":"constant/float32/float64/rank3/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank1/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank1/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank1/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank1/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank2/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank2/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank2/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank2/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank3/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank3/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank3/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"expected":[2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05,2,0.125,-4,6.103515625e-05],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float16/rank3/bias_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank1/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank1/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank1/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank1/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank2/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank2/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank2/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank2/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank3/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank3/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank3/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"expected":[1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576,1048576,1.0078125,9.3132257461547852e-10,-1048576],"tolerance":0,"passed":true} CASE {"name":"constant/float64/bfloat16/rank3/bias_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank1/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank1/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank1/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank1/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank2/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank2/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank2/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank2/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank3/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank3/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank3/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000,70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float32/rank3/bias_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank1/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank1/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank1/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank1/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank1/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[1,1,1,1],"expected":[1,1,1,1],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank2/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank2/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank2/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank2/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank2/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[2,2,2,2],"expected":[2,2,2,2],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank3/bias_identity","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank3/unit_weight_control","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank3/unit_weight_invariance","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank3/bias_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"expected":[1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624,1125899906842624,1.0000000000009095,6.8422776578360209e-49,-1125899906842624],"tolerance":0,"passed":true} CASE {"name":"constant/float64/float64/rank3/bias_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[6,6,6,6],"expected":[6,6,6,6],"tolerance":0,"passed":true} CASE {"name":"gradient/constant_bias_vjp_identity","actual_dtype":"float32","expected_dtype":"float32","actual":["Infinity",1,0,"-Infinity"],"expected":[70000,1.000244140625,9.3132257461547852e-10,-70000],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float16/strided0/forward","actual_dtype":"float16","expected_dtype":"float16","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided0/unit_weight","actual_dtype":"float16","expected_dtype":"float16","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided0/input_jvp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided0/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided1/forward","actual_dtype":"float16","expected_dtype":"float16","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided1/unit_weight","actual_dtype":"float16","expected_dtype":"float16","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided1/input_jvp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float16/strided1/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/bfloat16/strided0/forward","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5078125,-0.5,"-Infinity","Infinity",1.5078125,-0.5,"-Infinity"],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/bfloat16/strided0/unit_weight","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5078125,-0.5,"-Infinity","Infinity",1.5078125,-0.5,"-Infinity"],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/bfloat16/strided0/input_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/bfloat16/strided0/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/bfloat16/strided1/forward","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5078125,-0.5,"-Infinity","Infinity",1.5078125,-0.5,"-Infinity"],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/bfloat16/strided1/unit_weight","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5078125,-0.5,"-Infinity","Infinity",1.5078125,-0.5,"-Infinity"],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/bfloat16/strided1/input_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/bfloat16/strided1/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float32/strided0/forward","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float32/strided0/unit_weight","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float32/strided0/input_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float32/strided0/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float32/strided1/forward","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float32/strided1/unit_weight","actual_dtype":"float16","expected_dtype":"float32","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float32/strided1/input_jvp","actual_dtype":"float16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float32/strided1/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float64/strided0/forward","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float64/strided0/unit_weight","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float64/strided0/input_jvp","actual_dtype":"float16","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float64/strided0/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float16/float64/strided1/forward","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float64/strided1/unit_weight","actual_dtype":"float16","expected_dtype":"float64","actual":["Infinity",1.5,-0.5,"-Infinity","Infinity",1.5,-0.5,"-Infinity"],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float64/strided1/input_jvp","actual_dtype":"float16","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float16/float64/strided1/input_vjp","actual_dtype":"float16","expected_dtype":"float16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/float16/strided0/forward","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float16/strided0/unit_weight","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float16/strided0/input_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float16/strided0/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/float16/strided1/forward","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float16/strided1/unit_weight","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.5,1.5,0.625,-4.5,0.5],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float16/strided1/input_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float16/strided1/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided0/forward","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"expected":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided0/unit_weight","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"expected":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided0/input_jvp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided0/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided1/forward","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"expected":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided1/unit_weight","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"expected":[1048576,1.5078125,-0.5,-1048576,1048576,1.5078125,-0.5,-1048576],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided1/input_jvp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/bfloat16/strided1/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/float32/strided0/forward","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1.5,-0.5,-70144,70144,1.5,-0.5,-70144],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float32/strided0/unit_weight","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1.5,-0.5,-70144,70144,1.5,-0.5,-70144],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float32/strided0/input_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float32/strided0/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/float32/strided1/forward","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1.5,-0.5,-70144,70144,1.5,-0.5,-70144],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float32/strided1/unit_weight","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[70144,1.5,-0.5,-70144,70144,1.5,-0.5,-70144],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float32/strided1/input_jvp","actual_dtype":"bfloat16","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float32/strided1/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/float64/strided0/forward","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float64/strided0/unit_weight","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float64/strided0/input_jvp","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float64/strided0/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/bfloat16/float64/strided1/forward","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float64/strided1/unit_weight","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float64/strided1/input_jvp","actual_dtype":"bfloat16","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/bfloat16/float64/strided1/input_vjp","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided0/forward","actual_dtype":"float32","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided0/unit_weight","actual_dtype":"float32","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided0/input_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided0/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided1/forward","actual_dtype":"float32","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided1/unit_weight","actual_dtype":"float32","expected_dtype":"float32","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided1/input_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float16/strided1/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided0/forward","actual_dtype":"float32","expected_dtype":"float32","actual":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided0/unit_weight","actual_dtype":"float32","expected_dtype":"float32","actual":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided0/input_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided0/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided1/forward","actual_dtype":"float32","expected_dtype":"float32","actual":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided1/unit_weight","actual_dtype":"float32","expected_dtype":"float32","actual":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"expected":[1048575.5,1.5078125,-0.5,-1048575.5,1048575.5,1.5078125,-0.5,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided1/input_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/bfloat16/strided1/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided0/forward","actual_dtype":"float32","expected_dtype":"float32","actual":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided0/unit_weight","actual_dtype":"float32","expected_dtype":"float32","actual":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided0/input_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided0/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided1/forward","actual_dtype":"float32","expected_dtype":"float32","actual":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided1/unit_weight","actual_dtype":"float32","expected_dtype":"float32","actual":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"expected":[69999.5,1.500244140625,-0.5,-69999.5,69999.5,1.500244140625,-0.5,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided1/input_jvp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float32/strided1/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float64/strided0/forward","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float32/float64/strided0/unit_weight","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float32/float64/strided0/input_jvp","actual_dtype":"float32","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float32/float64/strided0/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float32/float64/strided1/forward","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float32/float64/strided1/unit_weight","actual_dtype":"float32","expected_dtype":"float64","actual":[1125899906842624,1.5,-0.5,-1125899906842624,1125899906842624,1.5,-0.5,-1125899906842624],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float32/float64/strided1/input_jvp","actual_dtype":"float32","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":false} CASE {"name":"nonconstant/float32/float64/strided1/input_vjp","actual_dtype":"float32","expected_dtype":"float32","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided0/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided0/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided0/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided0/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided1/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided1/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"expected":[1.5,0.625,-4.5,0.50006103515625,1.5,0.625,-4.5,0.50006103515625],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided1/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float16/strided1/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided0/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"expected":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided0/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"expected":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided0/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided0/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided1/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"expected":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided1/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"expected":[1048575.5,1.5078125,-0.49999999906867743,-1048575.5,1048575.5,1.5078125,-0.49999999906867743,-1048575.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided1/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/bfloat16/strided1/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided0/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"expected":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided0/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"expected":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided0/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided0/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided1/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"expected":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided1/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"expected":[69999.5,1.500244140625,-0.49999999906867743,-69999.5,69999.5,1.500244140625,-0.49999999906867743,-69999.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided1/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float32/strided1/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided0/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided0/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided0/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided0/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided1/forward","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided1/unit_weight","actual_dtype":"float64","expected_dtype":"float64","actual":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"expected":[1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5,1125899906842623.5,1.5000000000009095,-0.5,-1125899906842623.5],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided1/input_jvp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"expected":[0.34375,-0.09375,-0.15625,-0.09375,0.34375,-0.09375,-0.15625,-0.09375],"tolerance":0,"passed":true} CASE {"name":"nonconstant/float64/float64/strided1/input_vjp","actual_dtype":"float64","expected_dtype":"float64","actual":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"expected":[0.4375,-0.4375,1.4375,-1.4375,0.4375,-0.4375,1.4375,-1.4375],"tolerance":0,"passed":true} CASE {"name":"control/float16/no_params","actual_dtype":"float16","expected_dtype":"float16","actual":[-0.5,0.5,-0.5,0.5],"expected":[-0.5,0.5,-0.5,0.5],"tolerance":0,"passed":true} CASE {"name":"control/float16/weight_only","actual_dtype":"float16","expected_dtype":"float16","actual":[-1,2,1,-2],"expected":[-1,2,1,-2],"tolerance":0,"passed":true} CASE {"name":"control/float16/both","actual_dtype":"float16","expected_dtype":"float16","actual":[0,4,4,2],"expected":[0,4,4,2],"tolerance":0,"passed":true} CASE {"name":"control/float16/bias_only_same_dtype","actual_dtype":"float16","expected_dtype":"float16","actual":[0.5,2.5,2.5,4.5],"expected":[0.5,2.5,2.5,4.5],"tolerance":0,"passed":true} CASE {"name":"control/bfloat16/no_params","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[-0.5,0.5,-0.5,0.5],"expected":[-0.5,0.5,-0.5,0.5],"tolerance":0,"passed":true} CASE {"name":"control/bfloat16/weight_only","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[-1,2,1,-2],"expected":[-1,2,1,-2],"tolerance":0,"passed":true} CASE {"name":"control/bfloat16/both","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0,4,4,2],"expected":[0,4,4,2],"tolerance":0,"passed":true} CASE {"name":"control/bfloat16/bias_only_same_dtype","actual_dtype":"bfloat16","expected_dtype":"bfloat16","actual":[0.5,2.5,2.5,4.5],"expected":[0.5,2.5,2.5,4.5],"tolerance":0,"passed":true} CASE {"name":"control/float32/no_params","actual_dtype":"float32","expected_dtype":"float32","actual":[-0.5,0.5,-0.5,0.5],"expected":[-0.5,0.5,-0.5,0.5],"tolerance":0,"passed":true} CASE {"name":"control/float32/weight_only","actual_dtype":"float32","expected_dtype":"float32","actual":[-1,2,1,-2],"expected":[-1,2,1,-2],"tolerance":0,"passed":true} CASE {"name":"control/float32/both","actual_dtype":"float32","expected_dtype":"float32","actual":[0,4,4,2],"expected":[0,4,4,2],"tolerance":0,"passed":true} CASE {"name":"control/float32/bias_only_same_dtype","actual_dtype":"float32","expected_dtype":"float32","actual":[0.5,2.5,2.5,4.5],"expected":[0.5,2.5,2.5,4.5],"tolerance":0,"passed":true} CASE {"name":"control/float64/no_params","actual_dtype":"float64","expected_dtype":"float64","actual":[-0.5,0.5,-0.5,0.5],"expected":[-0.5,0.5,-0.5,0.5],"tolerance":0,"passed":true} CASE {"name":"control/float64/weight_only","actual_dtype":"float64","expected_dtype":"float64","actual":[-1,2,1,-2],"expected":[-1,2,1,-2],"tolerance":0,"passed":true} CASE {"name":"control/float64/both","actual_dtype":"float64","expected_dtype":"float64","actual":[0,4,4,2],"expected":[0,4,4,2],"tolerance":0,"passed":true} CASE {"name":"control/float64/bias_only_same_dtype","actual_dtype":"float64","expected_dtype":"float64","actual":[0.5,2.5,2.5,4.5],"expected":[0.5,2.5,2.5,4.5],"tolerance":0,"passed":true} SUMMARY {"checks":385,"failures":106}