Searched refs:kIntegerInput (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | subgraph_test_util.cc | 452 const int kIntegerInput = 2; in BuildLessEqualCondSubgraphWithDynamicTensor() local 468 ASSERT_EQ(subgraph->SetInputs({kStringInput1, kStringInput2, kIntegerInput}), in BuildLessEqualCondSubgraphWithDynamicTensor() 474 SetupTensor(subgraph, kIntegerInput, kTfLiteInt32); in BuildLessEqualCondSubgraphWithDynamicTensor() 482 subgraph->AddNodeWithParameters({kIntegerInput, kConstRhs}, {kOutput}, {}, in BuildLessEqualCondSubgraphWithDynamicTensor() 489 const int kIntegerInput = 2; in BuildBodySubgraphWithDynamicTensor() local 505 ASSERT_EQ(subgraph->SetInputs({kStringInput1, kStringInput2, kIntegerInput}), in BuildBodySubgraphWithDynamicTensor() 513 SetupTensor(subgraph, kIntegerInput, kTfLiteInt32); in BuildBodySubgraphWithDynamicTensor() 528 subgraph->AddNodeWithParameters({kIntegerInput, kConst}, {kIntegerOutput}, {}, in BuildBodySubgraphWithDynamicTensor() 541 const int kIntegerInput = 2; in BuildWhileSubgraphWithDynamicTensor() local 552 ASSERT_EQ(subgraph->SetInputs({kStringInput1, kStringInput2, kIntegerInput}), in BuildWhileSubgraphWithDynamicTensor() [all …]
|