Searched defs:col_align (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | lstm_fp16.c | 26 …tmWeightFp32ToFp16(float16_t *dst, const float *src, int batch, int deep, int col, int col_align) { in PackLstmWeightFp32ToFp16()
34 …LstmWeightFp16(float16_t *dst, const float16_t *src, int batch, int deep, int col, int col_align) { in PackLstmWeightFp16()
42 void PackLstmBiasFp32ToFp16(float16_t *dst, const float *src, int batch, int col, int col_align, in PackLstmBiasFp32ToFp16()
61 void PackLstmBiasFp16(float16_t *dst, const float16_t *src, int batch, int col, int col_align, bool… in PackLstmBiasFp16()
182 int row, int deep, int col, int col_align, bool is_vec) { in UpdateLstmGateFp16()
|
| D | matmul_fp16.c | 850 int col_align = UP_ROUND(col, C16NUM); in RowMajor2Row16MajorFp16Opt() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | lstm_fp32.c | 25 void PackLstmWeight(float *dst, const float *src, int batch, int deep, int col, int col_align) { in PackLstmWeight()
39 void PackLstmBias(float *dst, const float *src, int batch, int col, int col_align, bool is_bidirect… in PackLstmBias()
67 …t *c, const float *a, const float *b, const float *bias, int row, int deep, int col, int col_align, in LstmMatMul()
156 int col, int col_align, bool is_vec, float *packed_ptr) { in UpdateLstmGate()
|
| D | attention_fp32.c | 60 int col_align = vec_matmul ? real_col : UP_ROUND(real_col, col_tile); in RightMatrixPackElementSize() local
136 int col_align = vec_matmul ? real_col : UP_ROUND(real_col, col_tile); in PackRightMatrix() local
|
| D | matmul_fp32.c | 1073 int col_align) { in MatVecMulAvxFp32()
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/ |
| D | matmul_int8_wrapper.c | 34 … int col, int col_align, int deep_16, int input_zp, const int *weight_zp, const int *bias_ptr, in InitInt8MatrixB()
|
| /third_party/mesa3d/src/compiler/ |
| D | glsl_types.cpp | 2665 unsigned col_size, col_align; in get_explicit_type_for_size_align() local
|