Home
last modified time | relevance | path

Searched refs:SpaceToBatchNDOpConstModel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd_test.cc69 class SpaceToBatchNDOpConstModel : public SpaceToBatchNDOpModel { class
71 SpaceToBatchNDOpConstModel(const TensorData& input, in SpaceToBatchNDOpConstModel() function in tflite::__anonf57345680111::SpaceToBatchNDOpConstModel
114 SpaceToBatchNDOpConstModel({TensorType_FLOAT32, {1, 3, 3, 1}}, {2, 2}, in TEST()
121 SpaceToBatchNDOpConstModel m({TensorType_FLOAT32, {1, 4, 4, 1}}, {2, 2}, in TEST()
143 SpaceToBatchNDOpConstModel m({TensorType_FLOAT32, {2, 2, 4, 1}}, {2, 2}, in TEST()
165 SpaceToBatchNDOpConstModel m({TensorType_FLOAT32, {1, 5, 2, 1}}, {3, 2}, in TEST()
191 SpaceToBatchNDOpConstModel m({TensorType_FLOAT32, {1, 4, 2, 1}}, {3, 2}, in TEST()
231 EXPECT_DEATH(SpaceToBatchNDOpConstModel m( in TEST_F()
239 SpaceToBatchNDOpConstModel m({TensorType_UINT8, {1, 5, 2, 1}, -1.0, 1.0}, in TEST_F()
254 SpaceToBatchNDOpConstModel m({TensorType_INT8, {1, 5, 2, 1}, -1.0, 1.0}, in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1617 class SpaceToBatchNDOpConstModel : public SpaceToBatchNDOpModel { class
1619 SpaceToBatchNDOpConstModel(std::initializer_list<int> input_shape, in SpaceToBatchNDOpConstModel() function in tflite::__anon0c68fcb70111::SpaceToBatchNDOpConstModel
1635 SpaceToBatchNDOpConstModel m({1, 4, 4, 1}, {2, 2}, {0, 0, 0, 0}); in TEST()
1644 SpaceToBatchNDOpConstModel m({2, 2, 4, 1}, {2, 2}, {0, 0, 0, 0}); in TEST()
1653 SpaceToBatchNDOpConstModel m({1, 5, 2, 1}, {3, 2}, {1, 0, 2, 0}); in TEST()
1664 SpaceToBatchNDOpConstModel m({1, 4, 2, 1}, {3, 2}, {1, 1, 2, 4}); in TEST()