| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| H A D | quantization_mnist_test.py | 140 def _LoadWeights(self, model_dir, sess): argument
148 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): argument
190 def _Run(self, is_training, use_trt, batch_size, num_epochs, model_dir): argument
300 def _SaveModel(self, model_dir, output_dir): argument
323 def _GetFunc(self, use_trt, model_dir, use_dynamic_shape): argument
365 def _Run(self, use_trt, batch_size, model_dir, use_dynamic_shape=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| H A D | saved_model_api_test.cc | 75 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
107 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
205 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
309 std::string model_dir = SavedModelPath("AssetModule"); in TEST_P() local
364 std::string model_dir = SavedModelPath("StaticHashTableModule"); in TEST_P() local
497 std::string model_dir = tensorflow::io::JoinPath( in TEST_P() local
540 std::string model_dir = tensorflow::io::JoinPath( in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/experimental/tests/ |
| H A D | saved_model_api_test.cc | 61 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
85 std::string model_dir = SavedModelPath("VarsAndArithmeticObjectGraph"); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/kernel_test/ |
| H A D | input_generator.cc | 92 TfLiteStatus InputGenerator::LoadModel(const string& model_dir) { in LoadModel()
96 TfLiteStatus InputGenerator::LoadModel(const string& model_dir, in LoadModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A D | import_pb_to_tensorboard.py | 40 def import_to_tensorboard(model_dir, log_dir, tag_set): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | checkpoint_input_pipeline_hook_test.py | 56 def _read_vars(self, model_dir): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/ |
| H A D | fingerprinting.cc | 94 uint64 HashCheckpointIndexFile(absl::string_view model_dir) { in HashCheckpointIndexFile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/summary_ops/ |
| H A D | summary_ops_test.py | 1061 def __init__(self, model_dir): argument
1112 def __init__(self, model_dir): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/inference/ |
| H A D | server.py | 167 model_dir=".", argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | sharded_variable_test.py | 64 model_dir, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/ |
| H A D | saved_model_test.py | 574 model_dir, argument
|
| /aosp_15_r20/test/mlts/benchmark/tools/ |
| D | tensor_utils.py | 200 def __init__(self, model_dir): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | summary_ops_v2.py | 1149 def eval_dir(model_dir, name=None): argument
|