Searched refs:rsck (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
72 auto output = rsck.flat<float>(); 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
86 LIBXSMM_VLA_DECL(4, const float, input, rsck, S, C, K); in copy_RSCK_to_custom()
|