Searched refs:oper_out (Results 1 – 4 of 4) sorted by relevance
566 TF_Output oper_out) { in TF_OperationOutputConsumers_wrapper() argument567 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper()569 TF_OperationOutputConsumers(oper_out, consumers.data(), num_consumers); in TF_OperationOutputConsumers_wrapper()
202 TF_Output oper_out);
430 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,
1096 TF_DataType TF_OperationOutputType(TF_Output oper_out) { in TF_OperationOutputType() argument1098 oper_out.oper->node.output_type(oper_out.index)); in TF_OperationOutputType()1157 int TF_OperationOutputNumConsumers(TF_Output oper_out) { in TF_OperationOutputNumConsumers() argument1159 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() argument1170 for (const auto* edge : oper_out.oper->node.out_edges()) { in TF_OperationOutputConsumers()1171 if (edge->src_output() == oper_out.index) { in TF_OperationOutputConsumers()