Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 23 for Hare (0.15 sec)

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

                     absl::Span<AbstractTensorHandle* const> grad_outputs,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
        /* Given upstream grad U and a BiasAdd: A + bias, the gradients are:
         *
         *    dA = U
         *    dbias = reduceSum(U, dims = channel_dim)
         */
    
        AbstractTensorHandle* upstream_grad = grad_outputs[0];
        DCHECK(upstream_grad);
    
    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 bindings to the abstract classes defined in the
    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

    // `src_inputs[i]` will have that input replaced with `dst_inputs[i]`.  `prefix`
    // will be prepended to copied node names. `control_deps` are nodes in
    // `dst_graph` that the copied `src_graph` nodes will have control dependencies
    // on. `return_nodes` are nodes in `src_graph`, and the new corresponding nodes
    // in `dst_graph` will be returned. `return_nodes` must be non-null.
    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

        for (int i = 0; i < grad_outputs.size(); i++) {
          auto grad_input = grad_outputs[i];
          // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle
          // that takes care of this similar to `Tensor`?
          if (grad_input) {
            grad_input->Ref();
          }
          grad_inputs[i] = grad_input;
        }
        return absl::OkStatus();
      }
      ~IdentityNGradientFunction() 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/c_api_experimental.cc

      auto* optimizer_options =
          config.mutable_graph_options()->mutable_optimizer_options();
      if (enable) {
        optimizer_options->set_global_jit_level(tensorflow::OptimizerOptions::ON_1);
    
        // These XLA flags are needed to trigger XLA properly from C (more generally
        // non-Python) clients. If this API is called again with `enable` set to
        // false, it is safe to keep these flag values as is.
    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/experimental/filesystem/plugins/posix/posix_filesystem.cc

        return;
      } else if (S_ISDIR(st.st_mode)) {
        TF_SetStatus(status, TF_FAILED_PRECONDITION, "source path is a directory");
        return;
      }
    
      // Do the actual rename. Here both arguments are filenames.
      if (rename(src, dst) != 0)
        TF_SetStatusFromIOError(status, errno, dst);
      else
        TF_SetStatus(status, TF_OK, "");
    }
    
    static void CopyFile(const TF_Filesystem* filesystem, const char* src,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Mar 24 20:08:23 GMT 2024
    - 15.8K bytes
    - Viewed (0)
  7. tensorflow/c/checkpoint_reader.cc

        string key(v2_reader_->key());
        (*var_to_shape_map)[key] = TensorShape(entry.shape());
        (*var_to_data_type_map)[key] = DataType(entry.dtype());
      }
      // The returned pointers are owned by the caller.
      return std::make_pair(std::move(var_to_shape_map),
                            std::move(var_to_data_type_map));
    }
    
    }  // namespace checkpoint
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_unified_experimental_test.cc

    #include "tensorflow/core/platform/test.h"
    
    using tensorflow::Status;
    using tensorflow::string;
    using tensorflow::TF_StatusPtr;
    
    namespace tensorflow {
    namespace {
    
    // The tests are parameterized on:
    // - a string representing the tracing implementation: "mlir" or "graphdef".
    // - a boolean that when true enables TFRT as the execution engine.
    class UnifiedCAPI
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/filesystem/modular_filesystem.cc

    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/platform/env.h"
    #include "tensorflow/core/platform/file_system_helper.h"
    #include "tsl/platform/errors.h"
    
    // TODO(b/139060984): After all filesystems are converted, all calls to
    // methods from `FileSystem` will have to be replaced to calls to private
    // methods here, as part of making this class a singleton and the only way to
    // register/use filesystems.
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 23.1K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_test.cc

      EXPECT_TRUE(absl::StrContains(s, expected))
          << "'" << s << "' does not contain '" << expected << "'";
    }
    
    // Returns the GPU device name if there is one (with arbitrary tie breaking if
    // there are more than one), or "" otherwise.
    string GPUDeviceName(TF_Session* session) {
      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      TF_Status* s = status.get();
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top