Searched defs:_check_layer_norm_inputs (Results 1 – 3 of 3) sorted by relevance
11 C10_ALWAYS_INLINE std::pair<int64_t, int64_t> _check_layer_norm_inputs( in _check_layer_norm_inputs() function
12 void _check_layer_norm_inputs( in _check_layer_norm_inputs() function
472 C10_ALWAYS_INLINE void _check_layer_norm_inputs( in _check_layer_norm_inputs() function