Searched refs:kIntegerOutput (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | subgraph_test_util.cc | 492 const int kIntegerOutput = 5; in BuildBodySubgraphWithDynamicTensor() local 508 subgraph->SetOutputs({kStringOutput1, kStringOutput2, kIntegerOutput}), in BuildBodySubgraphWithDynamicTensor() 516 SetupTensor(subgraph, kIntegerOutput, kTfLiteInt32); in BuildBodySubgraphWithDynamicTensor() 528 subgraph->AddNodeWithParameters({kIntegerInput, kConst}, {kIntegerOutput}, {}, in BuildBodySubgraphWithDynamicTensor() 533 subgraph->AddNodeWithParameters({kIntegerOutput, kStringInput1}, in BuildBodySubgraphWithDynamicTensor() 544 const int kIntegerOutput = 5; in BuildWhileSubgraphWithDynamicTensor() local 555 subgraph->SetOutputs({kStringOutput1, kStringOutput2, kIntegerOutput}), in BuildWhileSubgraphWithDynamicTensor() 563 SetupTensor(subgraph, kIntegerOutput, kTfLiteInt32); in BuildWhileSubgraphWithDynamicTensor() 575 {kStringOutput1, kStringOutput2, kIntegerOutput}, {}, nullptr, 0, params, in BuildWhileSubgraphWithDynamicTensor()
|