- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TracingOperation (0.11 sec)
-
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) -
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) -
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) -
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) -
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)