Searched refs:primary_tile (Results 1 – 4 of 4) sorted by relevance
| /external/XNNPACK/tools/ |
| D | generate-argmaxpool-test.py | 36 primary_tile = int(match.group(3))
39 primary_tile = int(match.group(4))
45 return primary_tile, incremental_tile, channel_tile, arch, isa
1015 def generate_test_cases(ukernel, primary_tile, incremental_tile, channel_tile, argument
1042 "PRIMARY_TILE": primary_tile,
1080 primary_tile, incremental_tile, channel_tile, arch, isa = \
1086 test_case = generate_test_cases(name, primary_tile, incremental_tile,
|
| D | generate-maxpool-test.py | 34 primary_tile = int(match.group(2))
39 return primary_tile, incremental_tile, channel_tile, arch, isa
1002 def generate_test_cases(ukernel, primary_tile, incremental_tile, channel_tile, argument
1029 "PRIMARY_TILE": primary_tile,
1070 primary_tile, incremental_tile, channel_tile, arch, isa = \
1076 test_case = generate_test_cases(name, primary_tile, incremental_tile,
|
| /external/XNNPACK/test/ |
| D | argmaxpool-microkernel-tester.h | 79 inline ArgMaxPoolMicrokernelTester& pooling_tile(size_t primary_tile) { in pooling_tile() argument
80 assert(primary_tile != 0); in pooling_tile()
81 this->primary_pooling_tile_ = primary_tile; in pooling_tile()
86 inline ArgMaxPoolMicrokernelTester& pooling_tile(size_t primary_tile, size_t incremental_tile) { in pooling_tile() argument
87 assert(primary_tile != 0); in pooling_tile()
88 this->primary_pooling_tile_ = primary_tile; in pooling_tile()
|
| D | maxpool-microkernel-tester.h | 81 inline MaxPoolMicrokernelTester& pooling_tile(size_t primary_tile, size_t incremental_tile) { in pooling_tile() argument
82 assert(primary_tile != 0); in pooling_tile()
83 this->primary_pooling_tile_ = primary_tile; in pooling_tile()
|