Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc452 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 …]