- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for FunctionDef (0.19 sec)
-
tensorflow/c/eager/graph_function.h
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 // FunctionDef. Status GetFunctionDef(const FunctionDef** fdef) override; // Returns a shared reference to the wrapped function.
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
// Returns which subclass is this instance of. AbstractFunctionKind getKind() const { return kind_; } // Returns the AbstractFunction as a FunctionDef. virtual Status GetFunctionDef(const FunctionDef**) = 0; // Returns a shared reference to the wrapped function. virtual absl::StatusOr<core::RefCountPtr<FunctionRecord>> GetFunctionRecord() = 0; private: const AbstractFunctionKind kind_; }; using AbstractFunctionPtr =
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/graph_function.cc
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()); return absl::OkStatus(); } } // namespace graph
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/abstract_op_attrs.h
public: // Returns which subclass is this instance of. AbstractOpAttrsKind getKind() const { return kind_; } virtual ~AbstractOpAttrs() = default; // Returns the AbstractFunction as a FunctionDef. virtual void GetNameAttrList( tensorflow::NameAttrList* name_and_attrs) const = 0; virtual bool GetInt(absl::string_view, int64_t* result) const = 0;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed May 26 22:20:27 GMT 2021 - 2K bytes - Viewed (0) -
tensorflow/c/eager/abstract_context.h
// available to be called/referenced by its name in this context. virtual Status RegisterFunction(AbstractFunction*) = 0; // Remove a function. 'func' argument is the name of a previously added // FunctionDef. The name is in fdef.signature.name. virtual Status RemoveFunction(const string& func) = 0; private: const AbstractContextKind kind_; }; namespace internal { struct AbstractContextDeleter {
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:16:58 GMT 2021 - 3K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
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. // The FunctionDef only outputs the non-null tensors. We later pad the // function op outputs to have nullptrs at the `null_indices`. absl::flat_hash_set<int> null_indices; { AbstractContextPtr func_ctx(BuildFunction(fn_name));
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/c_test_util.h
bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def); bool GetNodeDef(TF_Operation* oper, tensorflow::NodeDef* node_def); bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def); bool GetAttrValue(TF_Operation* oper, const char* attr_name, tensorflow::AttrValue* attr_value, TF_Status* s); // Returns a sorted vector of std::pair<function_name, gradient_func> from
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 09 01:06:53 GMT 2018 - 6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.cc
#include "tensorflow/core/platform/test.h" #include "tensorflow/core/protobuf/tensorflow_server.pb.h" using ::tensorflow::string; string MatMulFunction(const string& matmul_device) { tensorflow::FunctionDef def; CHECK(tensorflow::protobuf::TextFormat::ParseFromString( absl::StrCat(" signature {" " name: 'MatMulFunction'" " input_arg {"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 9.1K bytes - Viewed (0)