[
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "constant/float16/bfloat16/rank1/bias_identity",
    "actual_dtype": "float16",
    "expected_dtype": "float32",
    "actual": [
      "Infinity",
      1.0078125,
      0,
      "-Infinity"
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/bfloat16/rank1/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/bfloat16/rank1/bias_jvp",
    "actual_dtype": "float16",
    "expected_dtype": "float32",
    "actual": [
      "Infinity",
      1.0078125,
      0,
      "-Infinity"
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/bfloat16/rank2/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/bfloat16/rank3/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "name": "constant/float16/float32/rank1/bias_identity",
    "actual_dtype": "float16",
    "expected_dtype": "float32",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float32/rank1/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float16/float32/rank1/unit_weight_invariance",
    "actual_dtype": "float16",
    "expected_dtype": "float32",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float32/rank1/bias_jvp",
    "actual_dtype": "float16",
    "expected_dtype": "float32",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float32/rank2/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float32/rank3/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "name": "constant/float16/float64/rank1/bias_identity",
    "actual_dtype": "float16",
    "expected_dtype": "float64",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float64/rank1/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float16/float64/rank1/unit_weight_invariance",
    "actual_dtype": "float16",
    "expected_dtype": "float64",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float64/rank1/bias_jvp",
    "actual_dtype": "float16",
    "expected_dtype": "float64",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float64/rank2/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float16/float64/rank3/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/bfloat16/rank1/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank1/unit_weight_control",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank1/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank1/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/bfloat16/rank2/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank2/unit_weight_control",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank2/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank2/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/bfloat16/rank3/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank3/unit_weight_control",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank3/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/bfloat16/rank3/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/float32/rank1/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float32/rank1/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/float32/rank1/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float32/rank1/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/float32/rank2/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float32/rank2/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/float32/rank2/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float32/rank2/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/float32/rank3/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float32/rank3/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/float32/rank3/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float32/rank3/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float32",
    "actual": [
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144,
      70144,
      1,
      9.313225746154785e-10,
      -70144
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "name": "constant/bfloat16/float64/rank1/bias_identity",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1,
      0,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float64/rank1/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/bfloat16/float64/rank1/unit_weight_invariance",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1,
      0,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float64/rank1/bias_jvp",
    "actual_dtype": "bfloat16",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1,
      0,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float64/rank2/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/bfloat16/float64/rank3/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "constant/float32/bfloat16/rank1/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank1/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank1/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank1/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float32/bfloat16/rank2/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank2/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank2/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank2/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float32/bfloat16/rank3/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank3/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank3/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/bfloat16/rank3/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float32/float32/rank1/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank1/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank1/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank1/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float32/float32/rank2/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank2/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank2/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank2/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float32/float32/rank3/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank3/unit_weight_control",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank3/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float32/rank3/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float32/float64/rank1/bias_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1,
      0,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float32/float64/rank1/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float32/float64/rank1/unit_weight_invariance",
    "actual_dtype": "float32",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1,
      0,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float32/float64/rank1/bias_jvp",
    "actual_dtype": "float32",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1,
      0,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float32/float64/rank2/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "name": "constant/float32/float64/rank3/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "constant/float64/bfloat16/rank1/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank1/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank1/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank1/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/bfloat16/rank2/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank2/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank2/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank2/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/bfloat16/rank3/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank3/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank3/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/bfloat16/rank3/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "expected": [
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576,
      1048576,
      1.0078125,
      9.313225746154785e-10,
      -1048576
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/float32/rank1/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank1/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank1/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank1/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/float32/rank2/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank2/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank2/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank2/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/float32/rank3/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank3/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank3/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float32/rank3/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000,
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/float64/rank1/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank1/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank1/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank1/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/float64/rank2/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank2/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank2/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank2/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "constant/float64/float64/rank3/bias_identity",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank3/unit_weight_control",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank3/unit_weight_invariance",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "constant/float64/float64/rank3/bias_jvp",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "expected": [
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624,
      1125899906842624,
      1.0000000000009095,
      6.842277657836021e-49,
      -1125899906842624
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "name": "gradient/constant_bias_vjp_identity",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      "Infinity",
      1,
      0,
      "-Infinity"
    ],
    "expected": [
      70000,
      1.000244140625,
      9.313225746154785e-10,
      -70000
    ],
    "tolerance": 0,
    "passed": false
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "nonconstant/float64/bfloat16/strided0/forward",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "expected": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "nonconstant/float64/bfloat16/strided0/unit_weight",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "expected": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "nonconstant/float64/bfloat16/strided1/forward",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "expected": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "nonconstant/float64/bfloat16/strided1/unit_weight",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "expected": [
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5,
      1048575.5,
      1.5078125,
      -0.4999999990686774,
      -1048575.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "nonconstant/float64/float32/strided0/forward",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "expected": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "nonconstant/float64/float32/strided0/unit_weight",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "expected": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "nonconstant/float64/float32/strided1/forward",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "expected": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "name": "nonconstant/float64/float32/strided1/unit_weight",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "expected": [
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5,
      69999.5,
      1.500244140625,
      -0.4999999990686774,
      -69999.5
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "control/float16/both",
    "actual_dtype": "float16",
    "expected_dtype": "float16",
    "actual": [
      0,
      4,
      4,
      2
    ],
    "expected": [
      0,
      4,
      4,
      2
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "control/bfloat16/both",
    "actual_dtype": "bfloat16",
    "expected_dtype": "bfloat16",
    "actual": [
      0,
      4,
      4,
      2
    ],
    "expected": [
      0,
      4,
      4,
      2
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "control/float32/both",
    "actual_dtype": "float32",
    "expected_dtype": "float32",
    "actual": [
      0,
      4,
      4,
      2
    ],
    "expected": [
      0,
      4,
      4,
      2
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  },
  {
    "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
  },
  {
    "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
  },
  {
    "name": "control/float64/both",
    "actual_dtype": "float64",
    "expected_dtype": "float64",
    "actual": [
      0,
      4,
      4,
      2
    ],
    "expected": [
      0,
      4,
      4,
      2
    ],
    "tolerance": 0,
    "passed": true
  },
  {
    "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
  }
]