- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for AddInput (0.09 sec)
-
tensorflow/c/eager/gradients.cc
forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } absl::Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInput(input)); forward_op_->inputs.push_back(input); return absl::OkStatus(); } absl::Status AddInputList(AbstractOperation* op_,
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/gradients_internal.h
absl::Status Reset(AbstractOperation*, const char* op, const char* raw_device_name, ForwardOperation*); // Records the inputs in the `ForwardOperation`. absl::Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*); absl::Status AddInputList(AbstractOperation*, absl::Span<AbstractTensorHandle* const> inputs,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 4.7K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
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(); string message = "This is the way!"; s = SetAttrString(check_numerics_op.get(), "message", message.data(),
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/abstract_operation.h
// // The value will override the previous value - that is, no "merging" of // existing and given constraints will be performed. virtual absl::Status SetDeviceName(const char* name) = 0; virtual absl::Status AddInput(AbstractTensorHandle* input) = 0; virtual absl::Status AddInputList( absl::Span<AbstractTensorHandle* const> inputs) = 0; virtual absl::Status Execute(absl::Span<AbstractTensorHandle*> retvals,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_AbstractTensor* const* inputs, TF_OutputList* o, TF_Status* s) { for (int i = 0; i < num_inputs; i++) { tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i]))); if (TF_GetCode(s) != TF_OK) { return; } } int num_outputs = unwrap(o)->expected_num_outputs; tsl::Set_TF_Status_from_Status( s, unwrap(op)->Execute(
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_graph.cc
absl::Status SetDeviceName(const char* name) override { // TODO(srbs): Implement this. device_name_ = name; return absl::OkStatus(); } absl::Status AddInput(AbstractTensorHandle* input) override { GraphTensor* t = dyn_cast<GraphTensor>(input); if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
return tensorflow::unwrap(op)->DeviceName().c_str(); } void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input)); } void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInputList(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)