Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for bitcast_op_test (0.14 sec)

  1. tensorflow/c/kernels/bitcast_op_test.cc

            << ctx.mutable_output(0)->shape().DebugString();
      }
    }
    
    TEST(BitcastOpTest, TestUpcast) {
      Tensor int8_input(DT_UINT8, {8});
      for (int i = 0; i < 8; i++) {
        int8_input.vec<uint8>()(i) = static_cast<uint8>(1);
      }
      TestBitcastOp(&int8_input, DT_UINT64, TensorShape(), error::OK);
    }
    
    TEST(BitcastOpTest, TestDowncast) {
      Tensor int64_input(static_cast<uint64>(1));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 18 15:10:51 UTC 2022
    - 5.5K bytes
    - Viewed (0)
  2. tensorflow/c/kernels/BUILD

        deps = [
            "//tensorflow/c:ops",
            "//tensorflow/c:tf_status",
            "//tensorflow/core:lib",
        ],
    )
    
    tf_cc_test(
        name = "bitcast_op_test",
        srcs = ["bitcast_op_test.cc"],
        deps = [
            ":bitcast_op",
            "//tensorflow/core:framework",
            "//tensorflow/core:protos_all_cc",
            "//tensorflow/core:test",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 27 18:00:18 UTC 2024
    - 4.8K bytes
    - Viewed (0)
Back to top