Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_NewWhile (0.18 sec)

  1. tensorflow/c/c_api.h

    // - Gradients
    // - Reference-type inputs
    // - Directly referencing external tensors from the cond/body graphs (this is
    //   possible in the Python API)
    TF_CAPI_EXPORT extern TF_WhileParams TF_NewWhile(TF_Graph* g, TF_Output* inputs,
                                                     int ninputs,
                                                     TF_Status* status);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  2. tensorflow/c/c_api.cc

          params.ninputs <= 0 || params.cond_inputs == nullptr ||
          params.body_inputs == nullptr || params.body_outputs == nullptr) {
        s->status = InvalidArgument(
            "TF_WhileParams must be created by successful TF_NewWhile() call");
        return false;
      }
      return true;
    }
    
    bool ValidateInputWhileParams(const TF_WhileParams& params, TF_Status* s) {
      if (params.cond_output.oper == nullptr) {
    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)
  3. tensorflow/c/c_api_function_test.cc

      // Add while loop to func_graph_
      {
        // The inputs to the while loop
        std::vector<TF_Output> inputs = {{feed1, 0}, {feed2, 0}};
        std::unique_ptr<TF_WhileParams> params(new TF_WhileParams(
            TF_NewWhile(func_graph_, &inputs[0], inputs.size(), s_)));
        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
        params->name = "test_loop";
    
        // Initialize outputs so we can easily detect errors/bugs
    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)
Back to top