- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for AddInputList (0.35 sec)
-
tensorflow/c/experimental/gradients/tape/tape_operation.h
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; Status SetAttrString(const char* attr_name, const char* data,
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
TF_RETURN_IF_ERROR(parent_op_->AddInput(input)); forward_op_.inputs.push_back(input); return OkStatus(); } Status TapeOperation::AddInputList( absl::Span<AbstractTensorHandle* const> inputs) { TF_RETURN_IF_ERROR(parent_op_->AddInputList(inputs)); for (auto input : inputs) { forward_op_.inputs.push_back(input); } return OkStatus(); }
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_internal.h
ForwardOperation*); // Records the inputs in the `ForwardOperation`. Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*); Status AddInputList(AbstractOperation*, absl::Span<AbstractTensorHandle* const> inputs, ForwardOperation*); // Sets the attrs in the `ForwardOperation`.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
TF_RETURN_IF_ERROR(op_->AddInput(input)); forward_op_->inputs.push_back(input); return absl::OkStatus(); } Status AddInputList(AbstractOperation* op_, absl::Span<AbstractTensorHandle* const> inputs, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInputList(inputs)); for (auto input : inputs) { forward_op_->inputs.push_back(input); } return absl::OkStatus(); }
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/c_test_util.cc
TF_Status* s, const char* name, TF_Operation** op, bool check) { TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name); TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; TF_AddInputList(desc, add_inputs, 2); *op = TF_FinishOperation(desc, s); if (check) { ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); ASSERT_NE(*op, nullptr); } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/gradients_test.cc
"or NotDifferentiableGradientFunction.", s.message()); ASSERT_EQ(nullptr, outputs[0]); } // TODO(b/164171226): Enable this test with tfrt after AddInputList is // supported. It is needed for IdentityN. #ifdef PLATFORM_GOOGLE INSTANTIATE_TEST_SUITE_P( UnifiedCAPI, CppGradients, ::testing::Combine(::testing::Values("graphdef", "mlir"),
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
if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } TF_AddInput(op_.get(), t->output_); return absl::OkStatus(); } Status AddInputList(absl::Span<AbstractTensorHandle* const> inputs) override { std::vector<TF_Output> tf_outputs(inputs.size()); for (int i = 0; i < inputs.size(); i++) { GraphTensor* t = dyn_cast<GraphTensor>(inputs[i]);
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) -
tensorflow/c/eager/abstract_operation.h
// 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, int* num_retvals) = 0;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0) -
tensorflow/c/c_api.h
// single tensor), and TF_AddInputList() for the second input (since // it takes a list, even if you were to pass a list with a single // tensor), as in: // TF_OperationDescription* desc = TF_NewOperation(graph, "Concat", "c"); // TF_Output concat_dim_input = {...}; // TF_AddInput(desc, concat_dim_input); // TF_Output values_inputs[5] = {{...}, ..., {...}}; // TF_AddInputList(desc, values_inputs, 5);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
tensorflow/c/eager/c_api.h
// TFE_DeleteOp() is called before TFE_DeleteContext(). // // Very similar to TF_OperationDescription with some differences: // (1) TF_Output or TFE_TensorHandle* as arguments to TF_AddInput, // TF_AddInputList // (2) TF_ColocateWith, TF_AddControlInput etc. do not make sense. // (3) Implementation detail: Avoid use of NodeBuilder/NodeDefBuilder since // the additional sanity checks there seem unnecessary;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1)