- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for AbstractFunction (0.3 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
TF_ExecutionContext* TF_CreateFunction(const char* fn_name, TF_Status* s) { return wrap(CreateTracingExecutionContext(fn_name, s)); } TF_AbstractFunction* TF_FinalizeFunction(TF_ExecutionContext* ctx, TF_OutputList* outputs, TF_Status* s) { AbstractFunction* func; TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(ctx)); if (!tracing_ctx) { tsl::Set_TF_Status_from_Status(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/graph_function.cc
#include "tensorflow/core/framework/function.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { namespace tracing { namespace graph { GraphFunction::GraphFunction(FunctionDef fdef) : AbstractFunction(kGraph), func_record_(new FunctionRecord(std::move(fdef), {}, true)) {} GraphFunction::~GraphFunction() {} Status GraphFunction::GetFunctionDef(const FunctionDef **fdef) { *fdef = &(func_record_->fdef());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Mar 04 19:49:06 GMT 2024 - 1.3K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs, bool use_function) { if (use_function) { const char* fn_name = "test_fn"; core::RefCountPtr<AbstractFunction> scoped_func; // Returning null tensors from a tf.function is not supported, so we keep // track of indices in the model's outputs are nullptr in this set.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_context.cc
// parent_ctx_->Unref(); } TapeOperation* TapeContext::CreateOperation() { return new TapeOperation(parent_ctx_->CreateOperation(), tape_, registry_); } Status TapeContext::RegisterFunction(AbstractFunction* f) { return parent_ctx_->RegisterFunction(f); } Status TapeContext::RemoveFunction(const string& func) { return parent_ctx_->RemoveFunction(func); } } // namespace gradients
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 1.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
} inputs_.push_back(t->output_); *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]); return absl::OkStatus(); } Status Finalize(OutputList* outputs, AbstractFunction** f) override { std::vector<TF_Output> graph_outputs; graph_outputs.reserve(outputs->outputs.size()); for (auto* abstract_output : outputs->outputs) {
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/c_api_unified_experimental_test.cc
ASSERT_EQ(h->FullType().type_id(), TFT_UNSET); ASSERT_EQ(unwrap(inputs[0])->FullType().type_id(), TFT_UNSET); // Clean up operation and inputs. TF_DeleteAbstractOp(add_op); TF_AbstractFunction* func = TF_FinalizeFunction(graph_ctx, add_outputs, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); // Note: TF_OutputList does not own the underlying AbstractTensors, those
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0)