Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for Hare (0.14 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/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)
  8. 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)
  9. 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)
  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