- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for AbstractFunction (0.18 sec)
-
tensorflow/c/eager/graph_function.h
#include "tensorflow/core/framework/function.h" #include "tensorflow/core/platform/refcount.h" namespace tensorflow { namespace tracing { namespace graph { using tensorflow::AbstractFunction; // Thin wrapper around a FunctionDef. class GraphFunction : public AbstractFunction { public: explicit GraphFunction(FunctionDef fdef); ~GraphFunction() override; // GraphFunction maybe stay alive for the duration of the returned
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Mar 04 19:49:06 GMT 2024 - 1.8K bytes - Viewed (0) -
tensorflow/c/eager/abstract_function.h
// function. class AbstractFunction : public core::RefCounted { protected: enum AbstractFunctionKind { kGraph, kMlir }; explicit AbstractFunction(AbstractFunctionKind kind) : kind_(kind) {} public: // Returns which subclass is this instance of. AbstractFunctionKind getKind() const { return kind_; } // Returns the AbstractFunction as a FunctionDef.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Mar 04 19:49:06 GMT 2024 - 1.9K bytes - Viewed (0) -
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/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)