Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 13 of 13 for ForwardOperation (0.19 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.h

      // For LLVM style RTTI.
      static bool classof(const AbstractOperation* ptr) {
        return ptr->getKind() == kTape;
      }
      ~TapeOperation() override;
    
     private:
      AbstractOperation* parent_op_;
      ForwardOperation forward_op_;
      Tape* tape_;
      const GradientRegistry& registry_;
    };
    
    }  // namespace gradients
    }  // namespace tensorflow
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 23 23:12:39 GMT 2020
    - 3.7K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradients_test.cc

        ASSERT_EQ(errors::OK, s.code()) << s.message();
        t.reset(x_raw);
      }
    
      AbstractOperationPtr check_numerics_op(ctx->CreateOperation());
      ForwardOperation forward_op;
      Status s = Reset(check_numerics_op.get(), "CheckNumerics",
                       /*raw_device_name=*/nullptr, &forward_op);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/tape/tape_operation.cc

                                  int* num_retvals) {
      TF_RETURN_IF_ERROR(parent_op_->Execute(retvals, num_retvals));
      for (int i = 0; i < *num_retvals; i++) {
        // TODO(srbs): Manage refcount of ForwardOperation's inputs/outputs.
        forward_op_.outputs.push_back(retvals[i]);
      }
      // TODO(b/166669239): This is needed to support AttrBuilder::Get for string
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
Back to top