Searched defs:shape_common (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | logsoftmax_quantized_test.cc | 38 const RuntimeShape& shape_common, in RunLogSoftmaxFloatReference()
71 const RuntimeShape& shape_common, in RunLogSoftmaxFloatReference()
102 const RuntimeShape& shape_common, in CheckOutputData()
147 const RuntimeShape& shape_common, int32 input_offset, in RunOneLogSoftmaxTest()
205 const RuntimeShape& shape_common, int32 input_offset, in RunOneLogSoftmaxTest()
264 auto shape_common = in TryOneUniformLogSoftmax() local
302 auto shape_common = in TryOneSkyscraperLogSoftmax() local
|
| H A D | softmax_quantized_test.cc | 36 const RuntimeShape& shape_common, in RunSoftmaxFloatReference()
66 const RuntimeShape& shape_common, in CheckOutputData()
107 const RuntimeShape& shape_common, int32 input_offset, in RunOneSoftmaxTest()
186 auto shape_common = in TryOneUniformSoftmax() local
224 auto shape_common = in TryOneSkyscraperSoftmax() local
|