- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for Consumer (0.19 sec)
-
tensorflow/c/c_api.cc
} return count; } int TF_OperationOutputConsumers(TF_Output oper_out, TF_Input* consumers, int max_consumers) { int count = 0; for (const auto* edge : oper_out.oper->node.out_edges()) { if (edge->src_output() == oper_out.index) { if (count < max_consumers) { consumers[count] = {ToOperation(edge->dst()), edge->dst_input()}; } ++count; }
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) -
tensorflow/c/eager/gradients.cc
~TapeVSpace() override {} // Returns the number of elements in the gradient tensor. int64_t NumElements(AbstractTensorHandle* tensor) const override; // Consumes references to the tensors in the gradient_tensors list and returns // a tensor with the result. AbstractTensorHandle* AggregateGradients( gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
EXPECT_EQ(attr_value.i(), 2); // Placeholder oper now has a consumer. ASSERT_EQ(1, TF_OperationOutputNumConsumers(TF_Output{feed, 0})); TF_Input feed_port; EXPECT_EQ(1, TF_OperationOutputConsumers(TF_Output{feed, 0}, &feed_port, 1)); EXPECT_EQ(add, feed_port.oper); EXPECT_EQ(0, feed_port.index); // The scalar const oper also has a consumer. ASSERT_EQ(1, TF_OperationOutputNumConsumers(TF_Output{three, 0}));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)