Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for flib_def (0.04 sec)

  1. 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)
  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);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
  3. 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)
  4. 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)
Back to top