Searched refs:CreateConstantInt32Tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 248 CreateConstantInt32Tensor(subgraph, kConstRhs, {1}, {rhs}); in BuildLessEqualCondSubgraph()
282 CreateConstantInt32Tensor(subgraph, kConstStep, {1}, {1}); in BuildAccumulateLoopBodySubgraph()
330 CreateConstantInt32Tensor(subgraph, kConstStep, {1}, {1}); in BuildPadLoopBodySubgraph()
333 CreateConstantInt32Tensor(subgraph, kConstPadding, {1, padding_dims}, in BuildPadLoopBodySubgraph()
406 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildAssignRandomValueToVariableSubgraph()
433 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildCallOnceAndReadVariableSubgraph()
480 CreateConstantInt32Tensor(subgraph, kConstRhs, {1}, {rhs}); in BuildLessEqualCondSubgraphWithDynamicTensor()
526 CreateConstantInt32Tensor(subgraph, kConst, {1}, {1}); in BuildBodySubgraphWithDynamicTensor()
579 void SubgraphBuilder::CreateConstantInt32Tensor(Subgraph* subgraph, in CreateConstantInt32Tensor() function in tflite::subgraph_test_util::SubgraphBuilder
|
| D | subgraph_test_util.h | 117 void CreateConstantInt32Tensor(Subgraph* subgraph, int tensor_index,
|