Searched refs:num_shapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | comparisons_test.cc | 815 const int num_shapes = 4; in TF_LITE_MICRO_TEST() local
817 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST()
820 for (int i = 0; i < num_shapes; ++i) { in TF_LITE_MICRO_TEST()
846 const int num_shapes = 4; in TF_LITE_MICRO_TEST() local
848 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST()
851 for (int i = 0; i < num_shapes; ++i) { in TF_LITE_MICRO_TEST()
877 const int num_shapes = 4; in TF_LITE_MICRO_TEST() local
879 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST()
882 for (int i = 0; i < num_shapes; ++i) { in TF_LITE_MICRO_TEST()
908 const int num_shapes = 4; in TF_LITE_MICRO_TEST() local
[all …]
|
| D | add_test.cc | 216 constexpr int num_shapes = 4; in TF_LITE_MICRO_TEST() local
218 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST()
225 for (int i = 0; i < num_shapes; ++i) { in TF_LITE_MICRO_TEST()
241 constexpr int num_shapes = 4; in TF_LITE_MICRO_TEST() local
243 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST()
250 for (int i = 0; i < num_shapes; ++i) { in TF_LITE_MICRO_TEST()
346 constexpr int num_shapes = 4; in TF_LITE_MICRO_TEST() local
348 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST()
364 for (int i = 0; i < num_shapes; i++) { in TF_LITE_MICRO_TEST()
378 constexpr int num_shapes = 4; in TF_LITE_MICRO_TEST() local
[all …]
|
| /external/libaom/libaom/test/ |
| D | av1_nn_predict_test.cc | 62 const int num_shapes);
64 const int num_shapes, const int run_times);
193 const int num_shapes) { in RunNnPredictTest_all() argument
194 for (int i = 0; i < num_shapes; i++) RunNnPredictTest(&shapes[i]); in RunNnPredictTest_all()
198 const int num_shapes, in RunNnPredictSpeedTest_all() argument
200 for (int i = 0; i < num_shapes; i++) in RunNnPredictSpeedTest_all()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.h | 367 int num_shapes);
378 const void* const* protos, const size_t* proto_lens, int num_shapes,
633 int num_shapes, int64_t* storage, int storage_size, TF_Status* status);
|
| D | c_api_experimental.h | 322 int num_shapes);
|
| D | c_api.cc | 922 int num_shapes) { in TF_SetAttrShapeList() argument
924 shapes.reserve(num_shapes); in TF_SetAttrShapeList()
925 for (int i = 0; i < num_shapes; ++i) { in TF_SetAttrShapeList()
961 const size_t* proto_lens, int num_shapes, in TF_SetAttrTensorShapeProtoList() argument
964 shapes.resize(num_shapes); in TF_SetAttrTensorShapeProtoList()
965 for (int i = 0; i < num_shapes; ++i) { in TF_SetAttrTensorShapeProtoList()
1422 int64_t** dims, int* num_dims, int num_shapes, in TF_OperationGetAttrShapeList() argument
1429 auto len = std::min(static_cast<int>(shapes.size()), num_shapes); in TF_OperationGetAttrShapeList()
|