Searched defs:tensor_count (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf_test.cc | 485 TfLiteTensor* tensors, const int tensor_count, in ValidateSVDFGoldens()
567 const int tensor_count = 6; // 5 inputs, 1 output in TestSVDF() local
627 const int tensor_count = 6; // 5 inputs, 1 output in TestIntegerSVDF() local
|
| D | shape_test.cc | 27 void ValidateShape(TfLiteTensor* tensors, const int tensor_count, in ValidateShape()
|
| D | logistic_test.cc | 50 void ValidateLogisticGoldens(TfLiteTensor* tensors, const int tensor_count, in ValidateLogisticGoldens()
|
| D | softmax_test.cc | 254 void ValidateSoftmaxGoldens(TfLiteTensor* tensors, const int tensor_count, in ValidateSoftmaxGoldens()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | memory_arena_threshold_test.cc | 92 size_t tensor_count = 0; member
|
| D | micro_allocator.cc | 156 AllocationInfoBuilder(AllocationInfo* info, size_t tensor_count, in AllocationInfoBuilder()
|