Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for TF_Function (0.08 sec)

  1. tensorflow/c/c_api_function.cc

      for (const Node* n : fn_body->graph.nodes()) {
        stack_traces[n->name()] = n->GetStackTrace();
      }
    
      TF_Function* tf_function = new TF_Function();
      tf_function->record = new tensorflow::FunctionRecord(
          std::move(fdef), std::move(stack_traces), false);
    
      return tf_function;
    }
    
    TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name,
    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

      TF_Operation* func_op = Use({func_feed});
      Run({{func_feed, Int32Tensor(3)}}, func_op, -3);
    }
    
    TEST_F(CApiFunctionTest, SameGradForTwoFunctions) {
      // Define the functions
      TF_Function* func1;
      TF_Function* func2;
      TF_Function* grad_func;
      DefineFunction("FooFunc1", &func1);
      DefineFunction("FooFunc2", &func2);
      DefineFunction("MyGrad", &grad_func);
    
      // Make grad_func be a gradient of func1 and func2
    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/c_api.h

      TF_Operation* oper;
      int index;  // The index of the output within oper.
    } TF_Output;
    
    // TF_Function is a grouping of operations with defined inputs and outputs.
    // Once created and added to graphs, functions can be invoked by creating an
    // operation whose operation type matches the function name.
    typedef struct TF_Function TF_Function;
    
    // Function definition options. TODO(iga): Define and implement
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental_test.cc

        TF_DeleteGraph(func_graph_);
        TF_DeleteStatus(s_);
      }
    
      const char* func_name_ = "MyFunc";
      TF_Status* s_;
      TF_Graph* func_graph_;
      TF_Function* func_;
    };
    
    TEST_F(CApiExperimentalFunctionTest, GraphRemoveFunction) {
      TF_Function* funcs[1];
      DefineFunction(func_name_, &func_);
    
      TF_GraphCopyFunction(func_graph_, func_, nullptr, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

      return ret;
    }
    
    char* TF_FunctionDebugString(TF_Function* func, size_t* len) {
      const auto& debug_str = DebugString(func->record->fdef());
      *len = debug_str.size();
      char* ret = static_cast<char*>(malloc(*len + 1));
      memcpy(ret, debug_str.c_str(), *len + 1);
      return ret;
    }
    
    // On success, returns a set of TF_Function instances from `text_proto` of
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.cc

        status->status = tensorflow::errors::NotFound(
            "Unable to find Function with name: ", name);
        return nullptr;
      }
    
      TF_Function* result = new TF_Function();
      record->Ref();
      result->record = record.get();
      return result;
    }
    
    void TFE_ContextRemoveFunction(TFE_Context* ctx, const char* name,
                                   TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_internal.h

      std::vector<tensorflow::string> missing_unused_key_names_data;
    };
    
    struct TF_DeviceList {
      std::vector<tensorflow::DeviceAttributes> response;
    };
    
    struct TF_Function {
      tensorflow::FunctionRecord* record;
    };
    
    struct TF_ApiDefMap {
      explicit TF_ApiDefMap(const tensorflow::OpList& op_list)
          :
    #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD)
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat May 13 00:49:12 UTC 2023
    - 7.6K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_experimental.h

    //
    // Do not return const char*, because some foreign language binding
    // (e.g. swift) cannot then call free() on the returned pointer.
    TF_CAPI_EXPORT extern char* TF_FunctionDebugString(TF_Function* func,
                                                       size_t* len);
    
    // On success, dequeues a tensor from a TF-managed FifoQueue given by
    // `tensor_id`, associated with `session`. There must be a graph node named
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_experimental.h

    TF_CAPI_EXPORT extern void TFE_ContextGetGraphDebugInfo(
        TFE_Context* ctx, const char* function_name, TF_Buffer* buf,
        TF_Status* status);
    
    // Extracts a TF_Function from the context.
    // Must call TF_DeleteFunction on the returned value.
    TF_CAPI_EXPORT extern TF_Function* TFE_ContextGetFunction(TFE_Context* ctx,
                                                              const char* name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api.h

    // TFE_Execute by creating an op with the same name as the function.
    TF_CAPI_EXPORT extern void TFE_ContextAddFunction(TFE_Context* ctx,
                                                      TF_Function* function,
                                                      TF_Status* status);
    
    // Removes a function from the context. Once removed, you can no longer
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
Back to top