Home
last modified time | relevance | path

Searched defs:tensor_info (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py30 def _log_tensor_details(tensor_info): argument
104 def gather_names(tensor_info): argument
/external/tensorflow/tensorflow/python/saved_model/
Dutils_impl.py143 def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
186 def get_element_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
Dsaved_model_test.py105 def _validate_inputs_tensor_info_fail(self, builder, tensor_info): argument
118 def _validate_inputs_tensor_info_accept(self, builder, tensor_info): argument
129 def _validate_outputs_tensor_info_fail(self, builder, tensor_info): argument
141 def _validate_outputs_tensor_info_accept(self, builder, tensor_info): argument
Dbuilder_impl.py155 def _validate_tensor_info(self, tensor_info): argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dgreedy_by_breadth_assignment.cc58 for (const auto& tensor_info : task_profiles[task_id]) { in GreedyByBreadthAssignment() local
73 for (const auto& tensor_info : task_profiles[task.task_id]) { in GreedyByBreadthAssignment() local
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto337 TensorInfo tensor_info = 1; field
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model.cc36 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, in GetTensorNamesFromTensorInfo()
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py97 def _get_element_from_tensor_info(tensor_info, graph): argument
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py405 def _print_tensor_info(tensor_info, indent=0): argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py576 def _gather_names(tensor_info): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3108 const auto& tensor_info = iter.second; in ParseInputArrays() local
3130 const auto& tensor_info = iter.second; in ParseOutputArrays() local