Searched refs:subgraph_input (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | tflite_subgraph_execute.cc | 116 TfLiteTensor* subgraph_input = in ResizeInputTensor() local
122 subgraph_input->dims->data[dim]) { in ResizeInputTensor()
146 TfLiteTensor* subgraph_input) { in SetSubgraphInput() argument
158 TfLiteTensorDataFree(subgraph_input); in SetSubgraphInput()
159 subgraph_input->data.raw = reinterpret_cast<char*>(tf_tensor_ptr); in SetSubgraphInput()
160 subgraph_input->bytes = required_bytes; in SetSubgraphInput()
161 subgraph_input->data_is_stale = true; in SetSubgraphInput()
162 subgraph_input->delegate = flex_delegate; in SetSubgraphInput()
167 TfLiteTensor* subgraph_input = in SetSubgraphInput() local
170 if (subgraph_input->type == kTfLiteString) { in SetSubgraphInput()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | if.cc | 93 TfLiteTensor* subgraph_input = subgraph->tensor(subgraph->inputs()[i]); in Prepare() local
95 SetTensorToDynamic(subgraph_input); in Prepare()
97 TF_LITE_ENSURE_TYPES_EQ(context, input->type, subgraph_input->type); in Prepare()
165 TfLiteTensor* subgraph_input = in Eval() local
168 if (IsDynamicTensor(subgraph_input)) { in Eval()
169 TfLiteTensorRealloc(input->bytes, subgraph_input); in Eval()
172 TF_LITE_ENSURE_EQ(context, input->bytes, subgraph_input->bytes); in Eval()
173 TfLiteTensorCopy(input, subgraph_input); in Eval()
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| D | call.cc | 61 TfLiteTensor* subgraph_input = subgraph->tensor(subgraph->inputs()[i]); in ValidateAndResizeInputsIfNeeded() local
63 TF_LITE_ENSURE_TYPES_EQ(context, node_input->type, subgraph_input->type); in ValidateAndResizeInputsIfNeeded()
69 if (!subgraph_input->dims->size) { in ValidateAndResizeInputsIfNeeded()
79 if (!MatchDimensionsExceptBatchSize(node_input, subgraph_input)) { in ValidateAndResizeInputsIfNeeded()
83 subgraph_input_dims << subgraph_input->dims->data[i] << " "; in ValidateAndResizeInputsIfNeeded()
91 subgraph_input->name, subgraph_input_dims.str().c_str()); in ValidateAndResizeInputsIfNeeded()
95 TF_LITE_ENSURE_EQ(context, subgraph_input->dims->data[0], 1); in ValidateAndResizeInputsIfNeeded()
|
| D | call_test.cc | 377 std::vector<int> subgraph_input = {kBatchSizeSubgraph, 7}; in TEST_F() local
384 subgraph->ResizeInputTensor(subgraph->inputs()[0], subgraph_input); in TEST_F()
385 subgraph->ResizeInputTensor(subgraph->inputs()[1], subgraph_input); in TEST_F()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | graph_fuser.cpp | 500 auto* subgraph_input = subgraph.inputs().at(input_index); in findFusedChunk() local
502 auto* node = subgraph_input->uses().at(0).user; in findFusedChunk()
504 AT_ASSERT(subgraph_input->uses().size() == 1); in findFusedChunk()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | python_function.cpp | 946 auto subgraph_input = subgraph->addInput(); in _append_subgraph() local
947 subgraph_input->copyMetadata(node->inputs().at(i)); in _append_subgraph()
948 value_map[node->inputs().at(i)] = subgraph_input; in _append_subgraph()
|