- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for flib_def (0.19 sec)
-
tensorflow/c/c_api_function.cc
int TF_GraphNumFunctions(TF_Graph* g) { tensorflow::mutex_lock l(g->mu); return g->graph.flib_def().num_functions(); } int TF_GraphGetFunctions(TF_Graph* g, TF_Function** funcs, int max_func, TF_Status* status) { tensorflow::FunctionDefLibrary lib; { tensorflow::mutex_lock l(g->mu); lib = g->graph.flib_def().ToProto(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/c_api_function_test.cc
const StackTracesMap* func_stack_traces; const StackTracesMap* grad_stack_traces; { mutex_lock l(host_graph_->mu); auto flib_def = host_graph_->graph.flib_def(); func_stack_traces = flib_def.GetStackTraces(func_name_); grad_stack_traces = flib_def.GetStackTraces("MyGrad"); } // Verify that stack traces of func is copied to graph function library. ASSERT_NE(func_stack_traces, nullptr);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6) -
tensorflow/c/c_api_experimental.cc
delete lib_handle; } void TF_GraphRemoveFunction(TF_Graph* g, const char* func_name, TF_Status* status) { tensorflow::mutex_lock l(g->mu); status->status = g->graph.mutable_flib_def()->RemoveFunction(func_name);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
const FunctionOptions& function_options, std::unique_ptr<tensorflow::Graph>* graph, tensorflow::FunctionLibraryDefinition* flib_def, std::vector<std::string>* control_ret_node_names, bool* control_rets_updated) override { // Inject failure to function instantiation if finding a node that contains
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/c_api.cc
if (node != nullptr && node->IsOp()) { NodeDef* const node_def = graph_def.add_node(); *node_def = node->def(); } } *graph_def.mutable_library() = graph.flib_def().ToProto(); if (flags::Global().more_stack_traces.value()) { *graph_def.mutable_debug_info() = graph.BuildDebugInfo(); } session->graph->mu.unlock();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0)