Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for ZerosLike (0.21 sec)

  1. tensorflow/c/eager/gradients.h

    // gradients.
    //
    // `ZerosLike` is not expected to be called and returns a nullptr. The creation
    // of default zeros grads is handled by the `DefaultGradientFunction` registered
    // for each op.
    // TODO(srbs): We need to define `ZerosLike` here to keep the compiler happy.
    // Figure out a way to avoid this.
    // TODO(srbs): Should ZerosLike check-fail instead of returning nullptr?
    class TapeTensor {
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Sep 26 10:27:05 GMT 2022
    - 6.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

    int64_t ToId(const AbstractTensorHandle* t) {
      return static_cast<int64_t>(reinterpret_cast<uintptr_t>(t));
    }
    
    Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* t,
                     AbstractTensorHandle** result) {
      AbstractOperationPtr op(ctx->CreateOperation());
      TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr));
      if (isa<tracing::TracingOperation>(op.get())) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/tape.h

        DCHECK(build_default_zeros_grads || zero_indices.empty());
        if (any_gradient_nonzero) {
          for (const auto i : zero_indices) {
            out_gradients[i] = trace.output_tensor_info[i].ZerosLike();
          }
          Status s;
          s = vspace.CallBackwardFunction(trace.op_type, trace.backward_function,
                                          unneeded_gradients, out_gradients,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  4. RELEASE.md

    *   Added support for Type III DCT, and `tf.spectral.idct(type=2|3)`.
    *   Correctly handle CuDNN RNN weight loaded when nest in `TimeDistributed`.
    *   Adding per-element weight support for `WALSComputePartialLhsAndRhsOp`.
    *   ZerosLike and OnesLike ops treated as constants by Graph Transform Tool.
    *   Gamma distribution and the derived distributions (Beta, Dirichlet, Student's
        t, inverse Gamma) now fully reparameterized.
    Plain Text
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Apr 03 20:27:38 GMT 2024
    - 727.4K bytes
    - Viewed (8)
Back to top