Home
last modified time | relevance | path

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

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