- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 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(); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 13.7K bytes - Viewed (0) -
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);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K 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
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 23.4K 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();
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 102.3K bytes - Viewed (0)