Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 28 for publish (0.4 sec)

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

    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_) {
          if (output) {
            output->Ref();
          }
    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/eager/c_api_unified_experimental.cc

      return nullptr;
    }
    
    }  // end namespace tracing
    }  // end namespace tensorflow
    
    // =============================================================================
    // Public C API entry points
    //
    // These are only the generic entry points for the C API. This file does not
    // have any visibility into the graph/eager implementation and is only providing
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

    #include "tensorflow/core/platform/stringpiece.h"
    #include "tensorflow/core/platform/thread_annotations.h"
    #include "tensorflow/core/platform/types.h"
    #include "tensorflow/core/public/session.h"
    #include "tensorflow/core/public/version.h"
    
    // The implementation below is at the top level instead of the
    // brain namespace because we are defining 'extern "C"' functions.
    using tensorflow::AttrValueMap;
    using tensorflow::DataType;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  4. 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,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
    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)
  5. tensorflow/c/experimental/gradients/custom_gradient_test.cc

        CHECK_EQ(errors::OK, s.code()) << s.message();
      }
    };
    
    class PassThroughGradientFunction : public GradientFunction {
     public:
      Status Compute(AbstractContext* ctx,
                     absl::Span<AbstractTensorHandle* const> grad_outputs,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
        CHECK_EQ(grad_outputs.size(), 1);
        CHECK_EQ(grad_inputs.size(), 1);
    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)
  6. tensorflow/c/eager/c_api_unified_experimental_eager.cc

    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h"
    #include "tensorflow/core/platform/strcat.h"
    
    // =============================================================================
    // Public C API entry points
    // These are only the entry points specific to the Eager API.
    // =============================================================================
    
    using tensorflow::AbstractContext;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jun 25 04:40:46 GMT 2020
    - 3.2K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/gradients/math_grad_test.cc

                         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());
        TF_SetTracingImplementation(std::get<0>(GetParam()), status.get());
    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)
  8. tensorflow/c/c_api_experimental.cc

      return ret;
    }
    
    void TF_MakeInternalErrorStatus(TF_Status* status, const char* errMsg) {
      status->status = tensorflow::errors::Internal(errMsg);
    }
    
    struct TF_CheckpointReader : public tensorflow::checkpoint::CheckpointReader {
      using tensorflow::checkpoint::CheckpointReader::CheckpointReader;
      std::vector<std::string> variable_list;
    };
    
    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)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    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)
  10. 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(
          const tensorflow::GraphOptimizationPassOptions& options) override {
        if (!enabled_) {
    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)
Back to top