Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for WithKernelLabel (0.22 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top