Searched refs:kcrs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d_test.cc | 68 LIBXSMM_INLINE void naive_copy_KCRS_to_RSCK(const float* kcrs, Tensor& rsck, in naive_copy_KCRS_to_RSCK() argument
70 LIBXSMM_VLA_DECL(4, const float, input, kcrs, C, R, S); in naive_copy_KCRS_to_RSCK()
|
| D | xsmm_conv2d.cc | 82 LIBXSMM_INLINE void copy_RSCK_to_custom(const float* rsck, float* kcrs, int R, in copy_RSCK_to_custom() argument
87 LIBXSMM_VLA_DECL(6, float, output, kcrs, blocksifm, R, S, ifmblock, ofmblock); in copy_RSCK_to_custom()
|