- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for AddInputList (0.12 sec)
-
tensorflow/c/eager/gradients_internal.h
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.cc
forward_op_->inputs.push_back(input); return absl::OkStatus(); } absl::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();
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_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"),
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
// 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, int* num_retvals) = 0;
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_graph.cc
if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } TF_AddInput(op_.get(), t->output_); return absl::OkStatus(); } absl::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]);
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
} void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInputList( {reinterpret_cast<tensorflow::AbstractTensorHandle**>( tensorflow::unwrap(inputs)), static_cast<size_t>(num_inputs)}); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)