Home
last modified time | relevance | path

Searched refs:input_details (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/lite/python/
Dlite_test.py168 input_details = interpreter.get_input_details()
169 self.assertLen(input_details, 1)
170 self.assertEqual('in_tensor', input_details[0]['name'])
171 self.assertEqual(np.float32, input_details[0]['dtype'])
172 self.assertAllEqual([2, 16, 16, 3], input_details[0]['shape'])
173 self.assertEqual((0., 0.), input_details[0]['quantization'])
242 input_details = interpreter.get_input_details()
243 self.assertLen(input_details, 1)
244 self.assertEqual('Placeholder', input_details[0]['name'])
245 self.assertEqual(np.float32, input_details[0]['dtype'])
[all …]
Dinterpreter_test.py106 input_details = interpreter.get_input_details()
107 self.assertEqual(1, len(input_details))
108 self.assertEqual('input', input_details[0]['name'])
109 self.assertEqual(np.float32, input_details[0]['dtype'])
110 self.assertTrue(([1, 4] == input_details[0]['shape']).all())
111 self.assertEqual((0.0, 0), input_details[0]['quantization'])
113 [], [], 0, input_details[0]['quantization_parameters'])
126 interpreter.set_tensor(input_details[0]['index'], test_input)
139 input_details = interpreter.get_input_details()
142 interpreter.set_tensor(input_details[0]['index'], test_input)
[all …]
Dconvert_test.py82 input_details = interpreter.get_input_details()
83 self.assertEqual(1, len(input_details))
84 self.assertEqual("input", input_details[0]["name"])
85 self.assertEqual(np.float32, input_details[0]["dtype"])
86 self.assertTrue(([1, 16, 16, 3] == input_details[0]["shape"]).all())
87 self.assertEqual((0., 0.), input_details[0]["quantization"])
121 input_details = interpreter.get_input_details()
122 self.assertEqual(2, len(input_details))
123 self.assertEqual("inputA", input_details[0]["name"])
124 self.assertEqual(np.uint8, input_details[0]["dtype"])
[all …]
Dlite_v2_test_util.py51 input_details = interpreter.get_input_details()
55 (input_details[idx]['shape_signature'] == shape_signature).all())
56 index = input_details[idx]['index']
61 input_details = interpreter.get_input_details()
63 for input_tensor, tensor_data in zip(input_details, input_data):
Dlite_flex_test.py68 input_details = interpreter.get_input_details()
70 interpreter.set_tensor(input_details[0]['index'], test_input)
100 input_details = interpreter.get_input_details()
102 interpreter.set_tensor(input_details[0]['index'], test_input)
135 input_details = interpreter.get_input_details()
137 interpreter.set_tensor(input_details[0]['index'], test_input)
217 input_details = interpreter.get_input_details()
219 interpreter.set_tensor(input_details[0]['index'], test_input)
Dlite_v2_test.py188 input_details = interpreter.get_input_details()
189 self.assertLen(input_details, 1)
190 self.assertEqual(np.float32, input_details[0]['dtype'])
270 input_details = interpreter.get_input_details()
271 self.assertLen(input_details, 1)
273 input_details[0]['dtype'])
329 input_details = interpreter.get_input_details()
330 self.assertLen(input_details, 1)
331 self.assertEqual(np.float32, input_details[0]['dtype'])
397 input_details = interpreter.get_input_details()
[all …]
Dutil_test.py125 input_details = interpreter.get_input_details()[0]
128 self.assertEqual(input_details["dtype"], np.uint32)
134 interpreter.set_tensor(input_details["index"], in_array)
362 input_details = interpreter.get_input_details()[0]
366 self.assertEqual(input_details["dtype"], in_tftype.as_numpy_dtype)
373 if input_details["dtype"] != np.float32:
375 scale, zero_point = input_details["quantization"]
377 input_data = input_data.astype(input_details["dtype"])
380 interpreter.set_tensor(input_details["index"], input_data)
/external/tensorflow/tensorflow/lite/examples/python/
Dlabel_image.py67 input_details = interpreter.get_input_details() variable
71 floating_model = input_details[0]['dtype'] == np.float32
74 height = input_details[0]['shape'][1]
75 width = input_details[0]['shape'][2]
84 interpreter.set_tensor(input_details[0]['index'], input_data)
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py89 input_details = the_interpreter.get_input_details()
92 if len(input_details) != len(input_ndarrays):
95 input_ndarrays, input_details, output_details))
96 for input_tensor, data in zip(input_details, input_ndarrays):
/external/tensorflow/tensorflow/lite/tutorials/
Dmnist_tflite.py59 input_details = interpreter.get_input_details()
63 input_image = np.reshape(input_image, input_details[0]['shape'])
64 interpreter.set_tensor(input_details[0]['index'], input_image)
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer_test.py133 input_details = interpreter.get_input_details()
135 interpreter.set_tensor(input_details[0]["index"], input_data)
176 input_details = interpreter.get_input_details()
178 interpreter.set_tensor(input_details[0]["index"], input_data)
214 input_details = interpreter.get_input_details()
215 interpreter.set_tensor(input_details[0]["index"], input_data)
/external/tensorflow/tensorflow/lite/experimental/quantization_debugger/
Ddebugger.py244 input_details = interpreter.get_input_details()
245 if len(input_details) != len(tensor_data):
248 'the model ({})'.format(len(tensor_data), len(input_details)))
251 for input_detail, tensor in zip(input_details, tensor_data):
255 for input_detail, tensor in zip(input_details, tensor_data):
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py207 input_details = interpreter.get_input_details()
209 detail["name"]: detail["index"] for detail in input_details
231 input_details = interpreter.get_input_details()
234 for input_tensor, data in zip(input_details, input_data)
266 input_details = interpreter.get_input_details()
275 for input_tensor in input_details:
312 input_details = interpreter.get_input_details()
315 for input_tensor, tensor_data in zip(input_details, input_data):
/external/tensorflow/tensorflow/lite/examples/experimental_new_converter/
Dkeras_lstm.ipynb190 "input_details = interpreter.get_input_details()\n",
192 "interpreter.set_tensor(input_details[0][\"index\"], x_test[0:1, :, :])\n",
DKeras_LSTM_fusion_Codelab.ipynb207 "input_details = interpreter.get_input_details()\n",
212 " interpreter.set_tensor(input_details[0][\"index\"], x_test[i:i+1, :, :])\n",
/external/tensorflow/tensorflow/lite/g3doc/models/style_transfer/
Doverview.ipynb314 " input_details = interpreter.get_input_details()\n",
315 " interpreter.set_tensor(input_details[0][\"index\"], preprocessed_style_image)\n",
353 " input_details = interpreter.get_input_details()\n",
357 " interpreter.set_tensor(input_details[0][\"index\"], preprocessed_content_image)\n",
358 " interpreter.set_tensor(input_details[1][\"index\"], style_bottleneck)\n",
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dpost_training_integer_quant.ipynb488 " input_details = interpreter.get_input_details()[0]\n",
497 " if input_details['dtype'] == np.uint8:\n",
498 " input_scale, input_zero_point = input_details[\"quantization\"]\n",
501 " test_image = np.expand_dims(test_image, axis=0).astype(input_details[\"dtype\"])\n",
502 " interpreter.set_tensor(input_details[\"index\"], test_image)\n",
/external/tensorflow/tensorflow/lite/micro/examples/micro_speech/train/
Dtrain_micro_speech_model.ipynb498 " input_details = interpreter.get_input_details()[0]\n",
503 " input_scale, input_zero_point = input_details[\"quantization\"]\n",
505 " test_data = test_data.astype(input_details[\"dtype\"])\n",
509 " interpreter.set_tensor(input_details[\"index\"], test_data[i])\n",
/external/tensorflow/tensorflow/lite/testing/
Dzip_test_utils.py391 input_details = interpreter.get_input_details()
393 for input_detail in input_details:
/external/tensorflow/tensorflow/lite/g3doc/models/super_resolution/
Doverview.ipynb220 "input_details = interpreter.get_input_details()\n",
224 "interpreter.set_tensor(input_details[0]['index'], lr)\n",
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dinference.md476 input_details = interpreter.get_input_details()
480 input_shape = input_details[0]['shape']
482 interpreter.set_tensor(input_details[0]['index'], input_data)
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3630 std::array<std::pair<int, string>, 2> input_details = { in TrySimplify() local
3633 for (const auto& input : input_details) { in TrySimplify()
/external/tensorflow/tensorflow/lite/micro/examples/hello_world/train/
Dtrain_hello_world_model.ipynb3062 " input_details = interpreter.get_input_details()[0]\n",
3066 " input_scale, input_zero_point = input_details[\"quantization\"]\n",
3069 " x_test_ = x_test_.astype(input_details[\"dtype\"])\n",
3074 " interpreter.set_tensor(input_details[\"index\"], [x_test_[i]])\n",