Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsubgraph_test_util.cc243 CreateConstantInt32Tensor(subgraph, kConstRhs, {1}, {rhs}); in BuildLessEqualCondSubgraph()
277 CreateConstantInt32Tensor(subgraph, kConstStep, {1}, {1}); in BuildAccumulateLoopBodySubgraph()
325 CreateConstantInt32Tensor(subgraph, kConstStep, {1}, {1}); in BuildPadLoopBodySubgraph()
328 CreateConstantInt32Tensor(subgraph, kConstPadding, {1, padding_dims}, in BuildPadLoopBodySubgraph()
401 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildAssignRandomValueToVariableSubgraph()
428 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildCallOnceAndReadVariableSubgraph()
464 CreateConstantInt32Tensor(subgraph, kConstResourceId, {1}, {1024}); in BuildCallOnceAndReadVariablePlusOneSubgraph()
465 CreateConstantInt32Tensor(subgraph, kConstOne, {1}, {1}); in BuildCallOnceAndReadVariablePlusOneSubgraph()
519 CreateConstantInt32Tensor(subgraph, kConstRhs, {1}, {rhs}); in BuildLessEqualCondSubgraphWithDynamicTensor()
565 CreateConstantInt32Tensor(subgraph, kConst, {1}, {1}); in BuildBodySubgraphWithDynamicTensor()
[all …]
H A Dsubgraph_test_util.h120 void CreateConstantInt32Tensor(Subgraph* subgraph, int tensor_index,