Home
last modified time | relevance | path

Searched refs:oper_out (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api.h430 TF_CAPI_EXPORT extern TF_DataType TF_OperationOutputType(TF_Output oper_out);
459 TF_CAPI_EXPORT extern int TF_OperationOutputNumConsumers(TF_Output oper_out);
468 TF_CAPI_EXPORT extern int TF_OperationOutputConsumers(TF_Output oper_out,
Dc_api.cc1096 TF_DataType TF_OperationOutputType(TF_Output oper_out) { in TF_OperationOutputType() argument
1098 oper_out.oper->node.output_type(oper_out.index)); in TF_OperationOutputType()
1157 int TF_OperationOutputNumConsumers(TF_Output oper_out) { in TF_OperationOutputNumConsumers() argument
1159 for (const auto* edge : oper_out.oper->node.out_edges()) { in TF_OperationOutputNumConsumers()
1160 if (edge->src_output() == oper_out.index) { in TF_OperationOutputNumConsumers()
1167 int TF_OperationOutputConsumers(TF_Output oper_out, TF_Input* consumers, in TF_OperationOutputConsumers() argument
1170 for (const auto* edge : oper_out.oper->node.out_edges()) { in TF_OperationOutputConsumers()
1171 if (edge->src_output() == oper_out.index) { in TF_OperationOutputConsumers()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc566 TF_Output oper_out) { in TF_OperationOutputConsumers_wrapper() argument
567 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper()
569 TF_OperationOutputConsumers(oper_out, consumers.data(), num_consumers); in TF_OperationOutputConsumers_wrapper()
Dtf_session_helper.h202 TF_Output oper_out);