Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for body_outputs (0.15 sec)

  1. tensorflow/c/while_loop_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      TF_Operation* add2 = Add(add1, one, params_->body_graph, s_, "add2");
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      params_->body_outputs[0] = {add2, 0};
      params_->body_outputs[1] = params_->body_inputs[1];
    
      // Finalize while loop
      ExpectOK();
    
      // Validate while loop outputs returned by TF_FinishWhile()
      EXPECT_TRUE(outputs_[0].oper != nullptr);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 11 06:05:56 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/while_gradients.cc

                                               std::vector<Output>* outputs) {
        std::vector<Output> body_outputs =
            ToOutputVector(while_ctx->body_outputs());
        std::vector<Output> body_inputs = ToOutputVector(while_ctx->body_inputs());
        return AddSymbolicGradients(scope, body_outputs, body_inputs, inputs,
                                    outputs);
      };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  3. tensorflow/cc/ops/while_loop.cc

      }
      TF_RETURN_IF_ERROR(scope.status());
    
      std::vector<Output> body_outputs;
      TF_RETURN_IF_ERROR(CreateBody(scope, body, switch_trues, &body_outputs));
    
      std::vector<Output> next_outputs(num_loop_vars);
      for (size_t i = 0; i < num_loop_vars; ++i) {
        next_outputs[i] = NextIteration(scope, body_outputs[i]);
        DCHECK_EQ(next_outputs[i].node()->name(), NextIterationName(scope, i));
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 01:01:21 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  4. tensorflow/c/c_api.cc

        s->status = InvalidArgument("TF_WhileParams `cond_output` field isn't set");
        return false;
      }
      for (int i = 0; i < params.ninputs; ++i) {
        if (params.body_outputs[i].oper == nullptr) {
          s->status = InvalidArgument("TF_WhileParams `body_outputs[", i, "]` ",
                                      "field isn't set");
          return false;
        }
      }
      if (params.name == nullptr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.h

    // variables.
    //
    // The returned TF_WhileParams will have all fields initialized except
    // `cond_output`, `body_outputs`, and `name`. The `body_outputs` buffer will be
    // allocated to size `ninputs`. The caller should build `cond_graph` and
    // `body_graph` starting from the inputs, and store the final outputs in
    // `cond_output` and `body_outputs`.
    //
    // If `status` is OK, the caller must call either TF_FinishWhile or
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_function_test.cc

        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
        TF_Operation* add2 = Add(add1, one, params->body_graph, s_, "add2");
        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
        params->body_outputs[0] = {add2, 0};
        params->body_outputs[1] = params->body_inputs[1];
    
        // Finalize while loop
        TF_FinishWhile(params.get(), s_, &outputs[0]);
        EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
Back to top