Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for SetOpName (0.22 sec)

  1. tensorflow/c/eager/tracing_utils.cc

    namespace tensorflow {
    namespace tracing {
    
    Status MaybeSetOpName(AbstractOperation* op, const char* op_name) {
      if (isa<TracingOperation>(op)) {
        TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name));
      }
      if (isa<gradients::TapeOperation>(op)) {
        TF_RETURN_IF_ERROR(MaybeSetOpName(
            dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(),
            op_name));
      }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        op_type_ = op;
        return absl::OkStatus();
      }
      Status SetOpName(const char* const op_name) override {
        if (op_) {
          return errors::FailedPrecondition(
              "SetOpName called on already built op.");
        }
        if (op_type_.empty()) {
          return errors::FailedPrecondition(
              "GraphOperation::Reset must be called before calling SetOpName.");
        }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  3. tensorflow/c/eager/gradients_test.cc

      ASSERT_EQ(errors::OK, s.code()) << s.message();
      if (isa<TracingOperation>(check_numerics_op.get())) {
        s = dyn_cast<TracingOperation>(check_numerics_op.get())
                ->SetOpName("check_numerics");
        ASSERT_EQ(errors::OK, s.code()) << s.message();
      }
      s = AddInput(check_numerics_op.get(), t.get(), &forward_op);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_internal.h

     public:
      // Sets the name of the operation: this is an optional identifier that is
      // not intended to carry semantics and preserved/propagated without
      // guarantees.
      virtual Status SetOpName(const char* op_name) = 0;
    
      // For LLVM style RTTI.
      static bool classof(const AbstractOperation* ptr) {
        return ptr->getKind() == kGraph || ptr->getKind() == kMlir;
      }
    };
    
    namespace internal {
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Nov 13 22:20:40 GMT 2020
    - 5.2K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_unified_experimental.cc

            s, tensorflow::errors::InvalidArgument(
                   "TF_AbstractOpSetOpName must be called on a TracingOperation."));
        return;
      }
      tsl::Set_TF_Status_from_Status(s, tracing_op->SetOpName(op_name));
    }
    
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s) {
      Status status =
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.cc

      TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr));
      if (isa<tracing::TracingOperation>(op.get())) {
        TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingOperation>(op.get())->SetOpName(
            absl::StrCat("ZerosLike", ToId(t)).c_str()));
      }
      TF_RETURN_IF_ERROR(op->AddInput(t));
      int num_outputs = 1;
      std::vector<AbstractTensorHandle*> outputs(num_outputs);
      TF_RETURN_IF_ERROR(
    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)
Back to top