Searched refs:kOutputValue (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | read_variable.cc | 31 constexpr int kOutputValue = 0; variable
45 GetOutputSafe(context, node, kOutputValue, &output)); in Prepare()
65 GetOutputSafe(context, node, kOutputValue, &output)); in Eval()
|
| D | subgraph_test_util.cc | 258 const int kOutputValue = 3; in BuildAccumulateLoopBodySubgraph() local
276 ASSERT_EQ(subgraph->SetOutputs({kOutputCounter, kOutputValue}), kTfLiteOk); in BuildAccumulateLoopBodySubgraph()
281 SetupTensor(subgraph, kOutputValue, kTfLiteInt32); in BuildAccumulateLoopBodySubgraph()
305 const int kOutputValue = 3; in BuildPadLoopBodySubgraph() local
323 ASSERT_EQ(subgraph->SetOutputs({kOutputCounter, kOutputValue}), kTfLiteOk); in BuildPadLoopBodySubgraph()
328 SetupTensor(subgraph, kOutputValue, kTfLiteInt32); in BuildPadLoopBodySubgraph()
349 subgraph->AddNodeWithParameters({kInputValue, kConstPadding}, {kOutputValue}, in BuildPadLoopBodySubgraph()
|