- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for TracingOperation (0.33 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 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K 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 23 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_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
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) -
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 23 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 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K 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 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1)