Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TracingOperation (0.11 sec)

  1. tensorflow/c/eager/c_api_unified_experimental.cc

                                TF_Status* s) {
      TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op));
      if (!tracing_op) {
        tsl::Set_TF_Status_from_Status(
            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));
    }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_internal.h

    // It is allowed to reusing the same abstract operation for multiple execution
    // on a given context, with the same or different input tensors.
    class TracingOperation : public AbstractOperation {
     protected:
      explicit TracingOperation(AbstractOperationKind kind)
          : AbstractOperation(kind) {}
    
     public:
      // Sets the name of the operation: this is an optional identifier that is
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients_test.cc

      absl::Status s = Reset(check_numerics_op.get(), "CheckNumerics",
                             /*raw_device_name=*/nullptr, &forward_op);
      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);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.cc

      AbstractOperationPtr op(ctx->CreateOperation());
      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;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      }
    
     private:
      TF_Graph* graph_;  // For shape inference.
    };
    
    // GraphOperation wraps and populates a TF_OperationDescription.
    class GraphOperation : public TracingOperation {
     public:
      explicit GraphOperation(TF_Graph* g) : TracingOperation(kGraph), g_(g) {}
      void Release() override { delete this; }
      absl::Status Reset(const char* op, const char* raw_device_name) override {
        if (op_) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
Back to top