HomeSort by relevance Sort by last modified time
    Searched refs:input_details (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/lite/python/
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...]
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...]
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...]
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/
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/
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/
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):

Completed in 734 milliseconds