Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for AddWithKernelLabel (0.32 sec)

  1. tensorflow/cc/framework/cc_ops_test.cc

      Scope root = Scope::NewRootScope();
      auto add = Add(root.WithKernelLabel("AddWithKernelLabel"), 1.0f, 2.0f);
      TF_EXPECT_OK(root.status());
      AttrSlice attrs = add.z.op().node()->attrs();
      const auto* kernel_attr = attrs.Find("_kernel");
      ASSERT_TRUE(kernel_attr);
      TF_EXPECT_OK(AttrValueHasType(*kernel_attr, "string"));
      EXPECT_EQ(kernel_attr->s(), "AddWithKernelLabel");
    }
    
    TEST(CCOpTest, ColocateWith) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 15 15:13:38 UTC 2023
    - 8.7K bytes
    - Viewed (0)
Back to top