Home
last modified time | relevance | path

Searched refs:test_model (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils_test.cc464 auto test_model = ReadConvModel(); in TEST_F() local
465 ASSERT_TRUE(test_model); in TEST_F()
466 auto readonly_model = test_model->GetModel(); in TEST_F()
513 auto test_model = ReadConvModel(); in TEST_F() local
514 ASSERT_TRUE(test_model); in TEST_F()
515 auto readonly_model = test_model->GetModel(); in TEST_F()
547 auto test_model = ReadConvModel(); in TEST_F() local
548 ASSERT_TRUE(test_model); in TEST_F()
549 auto readonly_model = test_model->GetModel(); in TEST_F()
579 auto test_model = ReadConvModel(); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/
Dsaved_model_error.py47 test_model = TestModule()
50 tf.saved_model.save(test_model, saved_model_path, options=save_options)
/external/libtextclassifier/native/actions/
Dtest-utils.cc70 void SetTestEntityDataSchema(ActionsModelT* test_model) { in SetTestEntityDataSchema() argument
73 test_model->actions_entity_data_schema.assign( in SetTestEntityDataSchema()
Dtest-utils.h33 void SetTestEntityDataSchema(ActionsModelT* test_model);
/external/tensorflow/tensorflow/lite/examples/experimental_new_converter/
Dstack_trace_example.py59 test_model = TestModule()
62 tf.saved_model.save(test_model, saved_model_path, options=save_options)
/external/tensorflow/tensorflow/python/keras/tests/
Dmodel_subclassing_test_util.py130 test_model = Inner()
131 x = test_model(x)
Dmodel_subclassing_test.py66 test_model = DummyModel()
68 test_model(dummy_data)
69 self.assertTrue(test_model.uses_custom_build, 'Model should use user '
/external/tensorflow/tensorflow/lite/
Dbuild_def.bzl403 …return [test_model for test_model in generated_test_models() if test_model not in generated_test_m…
572 for test_model in test_models[begin:end]:
573 tests_csv += "%s," % test_model
DBUILD341 "testdata/test_model.bin",
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py1347 def _TestVariablesGradient(self, inputs, test_model, vars_to_grad): argument
1350 test_model_re = custom_gradient.recompute_grad(test_model)
1355 out = test_model(inputs)
1381 test_model = TestKerasModelClass(10)
1384 test_model(test_input) # Ensures keras model is initialized.
1385 grads_re, grads = self._TestVariablesGradient(test_input, test_model,
1393 grads_re, grads = self._TestVariablesGradient(test_input, test_model,
1394 test_model.variables)
/external/tensorflow/tensorflow/lite/tools/
DBUILD137 "//tensorflow/lite:testdata/test_model.bin",
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_test.py1324 test_model = testing_utils.get_small_sequential_mlp(16, 2, input_dim=4)
1325 test_model.compile(loss='sparse_categorical_crossentropy',
1328 test_model.set_weights(fixed_weights)
1329 test_model_loss = test_model.train_on_batch(train_x, train_y)
1347 test_model = testing_utils.get_small_sequential_mlp(16, 2, input_dim=4)
1348 test_model.compile(loss='categorical_crossentropy',
1351 test_model.set_weights(fixed_weights)
1352 test_model_loss = test_model.train_on_batch(train_x, train_y)
1367 test_model = testing_utils.get_small_sequential_mlp(16, 1, input_dim=4)
1368 test_model.compile(loss='binary_crossentropy',
[all …]
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dkeras_test.py492 def test_model(self, member in KerasModelTest