- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TracingOperation (0.34 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)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/tracing_utils.cc
#include "tensorflow/core/platform/errors.h" 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(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 1.4K 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;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
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);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 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; } Status Reset(const char* op, const char* raw_device_name) override { if (op_) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1)