Searched defs:tensor_info (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 30 def _log_tensor_details(tensor_info): argument
104 def gather_names(tensor_info): argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| D | greedy_by_breadth_assignment.cc | 59 for (const auto& tensor_info : task_profiles[task_id]) { in GreedyByBreadthAssignment() local
74 for (const auto& tensor_info : task_profiles[task.task_id]) { in GreedyByBreadthAssignment() local
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | utils_impl.py | 149 def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
192 def get_element_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
|
| D | saved_model_test.py | 127 def _validate_inputs_tensor_info_fail(self, builder, tensor_info): argument
140 def _validate_inputs_tensor_info_accept(self, builder, tensor_info): argument
149 def _validate_outputs_tensor_info_fail(self, builder, tensor_info): argument
161 def _validate_outputs_tensor_info_accept(self, builder, tensor_info): argument
|
| D | builder_impl.py | 155 def _validate_tensor_info(self, tensor_info): argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | meta_graph.proto | 337 TensorInfo tensor_info = 1; field
|
| /external/tflite-support/tensorflow_lite_support/codegen/ |
| D | android_java_generator.cc | 110 TensorInfo tensor_info; in CreateTensorInfo() local
215 const TensorInfo& tensor_info) { in SetCodeWriterWithTensorInfo()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 36 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, in GetTensorNamesFromTensorInfo()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | wrap_function.py | 98 def _get_element_from_tensor_info(tensor_info, graph): argument
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 262 def _print_tensor_info(tensor_info, indent=0): argument
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 544 def _gather_names(tensor_info): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3582 TensorInfo tensor_info; in ConvertInitializer() local
3693 const auto& tensor_info = iter.second; in ParseInputArrays() local
|