- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for AddWithKernelLabel (0.55 sec)
-
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)