- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for body_outputs (0.15 sec)
-
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) -
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) -
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) -
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) -
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) -
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)