Searched refs:set_tensor_shapes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | util_test.py | 218 util.set_tensor_shapes([tensor], {"Placeholder": [5, 3, 5]})
225 util.set_tensor_shapes([tensor], {"Placeholder": [1, 3, 5]})
235 util.set_tensor_shapes([tensor], {"invalid-input": [5, 3, 5]})
248 util.set_tensor_shapes([tensor], {"Placeholder": [1, 5, 5]})
258 util.set_tensor_shapes([tensor], {})
|
| D | convert_saved_model.py | 204 util.set_tensor_shapes(in_tensors, input_shapes)
|
| D | util.py | 203 def set_tensor_shapes(tensors, shapes): function
|
| D | lite.py | 69 from tensorflow.lite.python.util import set_tensor_shapes as _set_tensor_shapes
|