- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for Hare (0.15 sec)
-
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) -
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) -
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) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
tensorflow::condition_variable finished_join_; // Temporary state between `StartExecute` and `Join`. // // Inputs; pointers are to objects not owned by the DeviceThread, but which // are expected to live at least until `Join` finishes: TFE_Context* context_ TF_GUARDED_BY(execution_mutex_); const char* operation_name_ TF_GUARDED_BY(execution_mutex_);
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) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// registration. // // Currently this is an error, and un-packing ParallelTensors must be performed // explicitly by running a TPUReplicatedOutput operation on the parallel device. // // TODO(allenl): There are some use-cases that are only supported by copying to // host at the moment (e.g. debug print on a tensor, .numpy(), etc.). We either // need to return something here or address these use-cases one by one.
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) -
tensorflow/c/eager/gradient_checker.cc
for (int i{}; i < inputs.size(); ++i) { theta_inputs[i] = inputs[i]; } AbstractTensorHandle* theta = theta_inputs[input_index]; // parameter we are grad checking // Convert from AbstractTensor to TF_Tensor. TF_Tensor* theta_tensor; TF_RETURN_IF_ERROR(GetValue(theta, &theta_tensor)); // Get number of elements and fill data.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
// Pack 3 variable handles into one TFE_TensorHandle. // When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3; std::vector<TFE_TensorHandle*> handles = {h0, h1, h2}; TFE_TensorHandle* packed_handle =
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) -
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) -
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) -
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)