- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for WithKernelLabel (1.48 sec)
-
tensorflow/cc/framework/cc_ops_test.cc
1e-5); test::ExpectTensorEqual<bool>(out[1], test::AsTensor<bool>({false}, {})); } TEST(CCOpTest, KernelLabel) { 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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 15 15:13:38 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/cc/framework/scope.h
/// of setting the status on the scope. Scope ExitOnError() const; /// Return a new scope. All ops created with the new scope will have /// kernel_label as the value for their '_kernel' attribute; Scope WithKernelLabel(const string& kernel_label) const; // The following functions are for scope object consumers. /// Return a unique name, using default_name if an op name has not been /// specified.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 09:08:33 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/cc/framework/scope.cc
/* clear_colocations */ true)); } Scope Scope::ExitOnError() const { return Scope(new Impl(*this, Impl::Tags::ExitOnError())); } Scope Scope::WithKernelLabel(const string& kernel_label) const { return Scope(new Impl(*this, Impl::Tags::KernelLabel(), kernel_label)); } CompositeOpScopes Scope::GetCompositeOpScopes( const string& composite_op_name) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 20.9K bytes - Viewed (0)