Searched refs:CreateConstantInt32Tensor (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | subgraph_test_util.cc | 242 CreateConstantInt32Tensor(subgraph, kConstRhs, {1}, {rhs}); in BuildLessEqualCondSubgraph() 276 CreateConstantInt32Tensor(subgraph, kConstStep, {1}, {1}); in BuildAccumulateLoopBodySubgraph() 324 CreateConstantInt32Tensor(subgraph, kConstStep, {1}, {1}); in BuildPadLoopBodySubgraph() 327 CreateConstantInt32Tensor(subgraph, kConstPadding, {1, padding_dims}, in BuildPadLoopBodySubgraph() 400 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildAssignRandomValueToVariableSubgraph() 427 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildCallOnceAndReadVariableSubgraph() 463 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildCallOnceAndReadVariablePlusOneSubgraph() 464 CreateConstantInt32Tensor(subgraph, kConstOne, {1}, {1}); in BuildCallOnceAndReadVariablePlusOneSubgraph() 518 CreateConstantInt32Tensor(subgraph, kConstRhs, {1}, {rhs}); in BuildLessEqualCondSubgraphWithDynamicTensor() 564 CreateConstantInt32Tensor(subgraph, kConst, {1}, {1}); in BuildBodySubgraphWithDynamicTensor() [all …]
|
D | subgraph_test_util.h | 118 void CreateConstantInt32Tensor(Subgraph* subgraph, int tensor_index,
|