Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc492 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()