Home
last modified time | relevance | path

Searched defs:subgraph_input (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dtflite_subgraph_execute.cc116 TfLiteTensor* subgraph_input = in ResizeInputTensor() local
146 TfLiteTensor* subgraph_input) { in SetSubgraphInput()
167 TfLiteTensor* subgraph_input = in SetSubgraphInput() local
/external/tensorflow/tensorflow/lite/kernels/
Dif.cc93 TfLiteTensor* subgraph_input = subgraph->tensor(subgraph->inputs()[i]); in Prepare() local
165 TfLiteTensor* subgraph_input = in Eval() local
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall.cc61 TfLiteTensor* subgraph_input = subgraph->tensor(subgraph->inputs()[i]); in ValidateAndResizeInputsIfNeeded() local
Dcall_test.cc377 std::vector<int> subgraph_input = {kBatchSizeSubgraph, 7}; in TEST_F() local
/external/pytorch/torch/csrc/jit/passes/
Dgraph_fuser.cpp500 auto* subgraph_input = subgraph.inputs().at(input_index); in findFusedChunk() local
/external/pytorch/torch/csrc/autograd/
Dpython_function.cpp946 auto subgraph_input = subgraph->addInput(); in _append_subgraph() local