- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for AddInput (0.18 sec)
-
tensorflow/c/c_test_util.cc
TF_AddInput(desc, l); TF_AddInput(desc, r); return TF_FinishOperation(desc, s); } TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype, TF_Graph* graph, TF_Status* s) { TF_OperationDescription* desc = TF_NewOperation(graph, "RandomUniform", "random_uniform"); TF_AddInput(desc, {shape, 0}); TF_SetAttrType(desc, "dtype", dtype);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
Status Reset(const char* op, const char* raw_device_name) override; const string& Name() const override; const string& DeviceName() const override; Status SetDeviceName(const char* name) override; Status AddInput(AbstractTensorHandle* input) override; Status AddInputList(absl::Span<AbstractTensorHandle* const> inputs) override; Status Execute(absl::Span<AbstractTensorHandle*> retvals, int* num_retvals) override;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
return parent_op_->DeviceName(); } Status TapeOperation::SetDeviceName(const char* name) { return parent_op_->SetDeviceName(name); } Status TapeOperation::AddInput(AbstractTensorHandle* input) { TF_RETURN_IF_ERROR(parent_op_->AddInput(input)); forward_op_.inputs.push_back(input); return OkStatus(); } Status TapeOperation::AddInputList( absl::Span<AbstractTensorHandle* const> inputs) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } 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(); } Status AddInputList(AbstractOperation* op_,
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/gradients_internal.h
// Records the op name in the `ForwardOperation`. Status Reset(AbstractOperation*, const char* op, const char* raw_device_name, ForwardOperation*); // Records the inputs in the `ForwardOperation`. Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*); Status AddInputList(AbstractOperation*, absl::Span<AbstractTensorHandle* const> inputs, ForwardOperation*);
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K 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 Status SetDeviceName(const char* name) = 0; virtual Status AddInput(AbstractTensorHandle* input) = 0; virtual Status AddInputList( absl::Span<AbstractTensorHandle* const> inputs) = 0; virtual Status Execute(absl::Span<AbstractTensorHandle*> retvals,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K 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(),
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_graph.cc
device_name_ = name; return absl::OkStatus(); } Status AddInput(AbstractTensorHandle* input) override { GraphTensor* t = dyn_cast<GraphTensor>(input); if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } TF_AddInput(op_.get(), t->output_); return absl::OkStatus(); }
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) -
tensorflow/c/eager/unified_api_testutil.cc
} AbstractOperationPtr fn_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); for (auto input : inputs) { TF_RETURN_IF_ERROR(fn_op->AddInput(input)); } int retvals = outputs.size() - null_indices.size(); std::vector<AbstractTensorHandle*> fn_outputs(retvals); TF_RETURN_IF_ERROR(fn_op->Execute(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K 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(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0)