Searched refs:set_tensor_shapes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model_test.py | 71 convert_saved_model.set_tensor_shapes([tensor], {"Placeholder": [5, 3, 5]})
79 convert_saved_model.set_tensor_shapes([tensor], {"Placeholder": [1, 3, 5]})
89 convert_saved_model.set_tensor_shapes([tensor],
103 convert_saved_model.set_tensor_shapes([tensor],
114 convert_saved_model.set_tensor_shapes([tensor], {})
|
| D | convert_saved_model.py | 193 def set_tensor_shapes(tensors, shapes): function
270 set_tensor_shapes(in_tensors, input_shapes)
|
| D | lite.py | 41 from tensorflow.lite.python.convert_saved_model import set_tensor_shapes as _set_tensor_shapes
|