Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 28 for class (0.18 sec)

  1. tensorflow/c/experimental/gradients/nn_grad.cc

    using std::vector;
    using tensorflow::ops::BiasAddGrad;
    using tensorflow::ops::Mul;
    using tensorflow::ops::ReluGrad;
    
    namespace tensorflow {
    namespace gradients {
    namespace {
    
    class ReluGradientFunction : public GradientFunction {
     public:
      explicit ReluGradientFunction(vector<AbstractTensorHandle*> f_outputs)
          : forward_outputs_(f_outputs) {
        for (auto output : forward_outputs_) {
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/custom_gradient_test.cc

    #include "tensorflow/core/platform/errors.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace gradients {
    namespace internal {
    namespace {
    using std::vector;
    
    class CustomGradientTest
        : public ::testing::TestWithParam<std::tuple<const char*, bool, bool>> {
     protected:
      void SetUp() override {
        TF_StatusPtr status(TF_NewStatus());
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 4.8K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/array_grad.cc

    #include "tensorflow/c/experimental/gradients/array_grad.h"
    
    #include "tensorflow/c/eager/abstract_context.h"
    
    namespace tensorflow {
    namespace gradients {
    namespace {
    class IdentityNGradientFunction : public GradientFunction {
     public:
      Status Compute(AbstractContext* ctx,
                     absl::Span<AbstractTensorHandle* const> grad_outputs,
    C++
    - Registered: Tue Apr 09 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 1.6K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/gradients/math_grad_test.cc

                         absl::Span<AbstractTensorHandle* const> inputs,
                         absl::Span<AbstractTensorHandle*> outputs) {
      return ops::DivNoNan(ctx, inputs[0], inputs[1], &outputs[0], "DivNoNan");
    }
    
    class CppGradients
        : public ::testing::TestWithParam<std::tuple<const char*, bool, bool>> {
     protected:
      void SetUp() override {
        TF_StatusPtr status(TF_NewStatus());
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Thu Apr 13 17:32:14 GMT 2023
    - 16.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

                                           TF_Status* status) {
      status->status = tensorflow::FindKernelDef(
          tensorflow::DeviceType(device_type), builder->BuildNodeDef(),
          /* def = */ nullptr, /* kernel_class_name = */ nullptr);
    }
    
    const char* TF_GetNumberAttrForOpListInput(const char* op_name, int input_index,
                                               TF_Status* status) {
      const tensorflow::OpDef* op_def = nullptr;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    namespace parallel_device {
    namespace {
    
    class OpDeleter {
     public:
      void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); }
    };
    
    using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
    
    class StatusDeleter {
     public:
      void operator()(TF_Status* to_delete) const { TF_DeleteStatus(to_delete); }
    };
    
    using StatusPtr = std::unique_ptr<TF_Status, StatusDeleter>;
    
    class ExecutorDeleter {
     public:
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  7. tensorflow/c/eager/parallel_device/parallel_device.cc

    #include "tensorflow/c/eager/tfe_tensorhandle_internal.h"
    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_status_helper.h"
    
    namespace tensorflow {
    namespace parallel_device {
    namespace {
    
    class OpDeleter {
     public:
      void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); }
    };
    
    using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
    
    using MaybeParallelTensorOwned =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/gradients_test.cc

    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace gradients {
    namespace internal {
    namespace {
    using std::vector;
    using tensorflow::TF_StatusPtr;
    using tracing::TracingOperation;
    
    class CppGradients
        : public ::testing::TestWithParam<std::tuple<const char*, bool, bool>> {
     protected:
      void SetUp() override {
        TF_StatusPtr status(TF_NewStatus());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_distributed_test.cc

          VariableAddFunctionSignature(), &def));
      return def.SerializeAsString();
    }
    
    // A graph optimization pass that would fail when triggered for more than once.
    class GraphErrorInjectionPass : public tensorflow::GraphOptimizationPass {
     public:
      static bool enabled_;
      GraphErrorInjectionPass() {}
    
      tensorflow::Status Run(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc

    }
    
    namespace tensorflow {
    namespace {
    
    // TODO(vnvo2409): Refactor `gcs_filesystem_test` to remove unnecessary tests
    // after porting all tests from
    // `//tensorflow/core/platform/cloud:gcs_file_system_test`.
    class GCSFilesystemTest : public ::testing::Test {
     public:
      void SetUp() override {
        root_dir_ = io::JoinPath(
            *GetTmpDir(),
            ::testing::UnitTest::GetInstance()->current_test_info()->name());
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Aug 31 12:04:23 GMT 2020
    - 24.9K bytes
    - Viewed (0)
Back to top