| /external/tensorflow/tensorflow/lite/kernels/ |
| D | maximum_minimum_test.cc | 75 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel() function
100 TestModel<float>(BuiltinOperator_MAXIMUM, {TensorType_FLOAT32, {3, 1, 2}}, in TEST()
104 TestModel<float>(BuiltinOperator_MINIMUM, {TensorType_FLOAT32, {3, 1, 2}}, in TEST()
113 TestModel<uint8_t>(BuiltinOperator_MAXIMUM, {TensorType_UINT8, {3, 1, 2}}, in TEST()
117 TestModel<uint8_t>(BuiltinOperator_MINIMUM, {TensorType_UINT8, {3, 1, 2}}, in TEST()
126 TestModel<int8_t>(BuiltinOperator_MAXIMUM, {TensorType_INT8, {3, 1, 2}}, in TEST()
129 TestModel<int8_t>(BuiltinOperator_MINIMUM, {TensorType_INT8, {3, 1, 2}}, in TEST()
137 TestModel<int16_t>(BuiltinOperator_MAXIMUM, {TensorType_INT16, {3, 1, 2}}, in TEST()
141 TestModel<int16_t>(BuiltinOperator_MINIMUM, {TensorType_INT16, {3, 1, 2}}, in TEST()
150 TestModel<float>(BuiltinOperator_MAXIMUM, {TensorType_FLOAT32, {3, 1, 2}}, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | min_max_builder_test.cc | 80 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel() function
120 TestModel<uint8_t>(BuiltinOperator_MAXIMUM, in TEST()
138 TestModel<uint8_t>(BuiltinOperator_MINIMUM, in TEST()
156 TestModel<int8_t>(BuiltinOperator_MAXIMUM, in TEST()
165 TestModel<int8_t>(BuiltinOperator_MINIMUM, in TEST()
|
| D | batch_seq_config_test.cc | 88 class TestModel { class
90 TestModel() : delegate_(nullptr, [](TfLiteDelegate* delegate) {}) {} in __anonbe5bfbe50302() function in tflite::TestModel
185 auto default_model = std::make_unique<TestModel>(); in TEST()
186 auto delegated_model = std::make_unique<TestModel>(); in TEST()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention_test.py | 271 class TestModel(keras.Model): class
274 super(TestModel, self).__init__()
298 model = TestModel()
312 model = TestModel()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_test.py | 3034 class TestModel(training_module.Model): class
3037 super(TestModel, self).__init__(name='test_model')
3049 model = TestModel()
3166 class TestModel(training_module.Model): class
3169 super(TestModel, self).__init__(name='test_model')
3177 model = TestModel()
3193 class TestModel(training_module.Model): class
3196 super(TestModel, self).__init__(name='test_model')
3208 model = TestModel()
3268 class TestModel(training_module.Model): class
[all …]
|
| D | base_layer_test.py | 1431 class TestModel(training_lib.Model): class
1434 super(TestModel, self).__init__(
1444 model = TestModel()
1461 class TestModel(training_lib.Model): class
1464 super(TestModel, self).__init__(
1475 model = TestModel()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models_test.py | 44 class TestModel(keras.Model): class
49 super(TestModel, self).__init__()
|