Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_OperationNumControlInputs (3.11 sec)

  1. tensorflow/c/c_api_test.cc

      TF_Operation* control_ops[100];
      EXPECT_EQ(0, TF_OperationNumControlInputs(scalar));
      EXPECT_EQ(0, TF_OperationGetControlInputs(scalar, control_ops, 100));
      EXPECT_EQ(0, TF_OperationNumControlOutputs(scalar));
      EXPECT_EQ(0, TF_OperationGetControlOutputs(scalar, control_ops, 100));
    
      EXPECT_EQ(0, TF_OperationNumControlInputs(feed));
      EXPECT_EQ(0, TF_OperationGetControlInputs(feed, control_ops, 100));
    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)
  2. tensorflow/c/c_api.cc

          if (count < max_consumers) {
            consumers[count] = {ToOperation(edge->dst()), edge->dst_input()};
          }
          ++count;
        }
      }
      return count;
    }
    
    int TF_OperationNumControlInputs(TF_Operation* oper) {
      int count = 0;
      for (const auto* edge : oper->node.in_edges()) {
        if (edge->IsControlEdge() && !edge->src()->IsSource()) {
          ++count;
        }
      }
      return 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)
Back to top