Searched refs:input_details (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_test.py | 103 input_details = interpreter.get_input_details()
104 self.assertEqual(1, len(input_details))
105 self.assertEqual('Placeholder', input_details[0]['name'])
106 self.assertEqual(np.float32, input_details[0]['dtype'])
107 self.assertTrue(([1, 16, 16, 3] == input_details[0]['shape']).all())
108 self.assertEqual((0., 0.), input_details[0]['quantization'])
132 input_details = interpreter.get_input_details()
133 self.assertEqual(1, len(input_details))
134 self.assertEqual('Placeholder', input_details[0]['name'])
135 self.assertEqual(np.string_, input_details[0]['dtype'])
[all …]
|
| D | interpreter_test.py | 38 input_details = interpreter.get_input_details()
39 self.assertEqual(1, len(input_details))
40 self.assertEqual('input', input_details[0]['name'])
41 self.assertEqual(np.float32, input_details[0]['dtype'])
42 self.assertTrue(([1, 4] == input_details[0]['shape']).all())
43 self.assertEqual((0.0, 0), input_details[0]['quantization'])
54 interpreter.set_tensor(input_details[0]['index'], test_input)
69 input_details = interpreter.get_input_details()
70 self.assertEqual(1, len(input_details))
71 self.assertEqual('input', input_details[0]['name'])
[all …]
|
| D | convert_test.py | 112 input_details = interpreter.get_input_details()
113 self.assertEqual(1, len(input_details))
114 self.assertEqual("input", input_details[0]["name"])
115 self.assertEqual(np.float32, input_details[0]["dtype"])
116 self.assertTrue(([1, 16, 16, 3] == input_details[0]["shape"]).all())
117 self.assertEqual((0., 0.), input_details[0]["quantization"])
149 input_details = interpreter.get_input_details()
150 self.assertEqual(2, len(input_details))
151 self.assertEqual("inputA", input_details[0]["name"])
152 self.assertEqual(np.uint8, input_details[0]["dtype"])
[all …]
|
| D | lite_v2_test.py | 45 input_details = interpreter.get_input_details()
48 for input_tensor, tensor_data in zip(input_details, input_data):
|
| /external/tensorflow/tensorflow/lite/examples/python/ |
| D | label_image.py | 54 input_details = interpreter.get_input_details() variable
58 if input_details[0]['dtype'] == np.float32:
62 height = input_details[0]['shape'][1]
63 width = input_details[0]['shape'][2]
73 interpreter.set_tensor(input_details[0]['index'], input_data)
|
| /external/tensorflow/tensorflow/lite/tutorials/ |
| D | mnist_tflite.py | 58 input_details = interpreter.get_input_details()
62 input_image = np.reshape(input_image, input_details[0]['shape'])
63 interpreter.set_tensor(input_details[0]['index'], input_image)
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib.py | 103 input_details = interpreter.get_input_details()
110 dtype=input_tensor["dtype"]) for input_tensor in input_details
127 input_details = interpreter.get_input_details()
130 for input_tensor, tensor_data in zip(input_details, input_data):
|
| /external/tensorflow/tensorflow/lite/g3doc/r2/convert/ |
| D | python_api.md | 140 input_details = interpreter.get_input_details()
144 input_shape = input_details[0]['shape']
146 interpreter.set_tensor(input_details[0]['index'], input_data)
|
| /external/tensorflow/tensorflow/lite/g3doc/convert/ |
| D | python_api.md | 199 input_details = interpreter.get_input_details()
203 input_shape = input_details[0]['shape']
205 interpreter.set_tensor(input_details[0]['index'], input_data)
|