Searched refs:input_alignment (Results 1 – 2 of 2) sorted by relevance
52 uint8_t input_alignment; member127 key.input_alignment = cudnn_utils::getAlignment(input); in apply_impl_helper()178 …ls::getTensorDescriptor(input.sizes(), input.strides(), CUDNN_DATA_INT8, 'x', key.input_alignment)) in apply_impl_helper()
50 uint8_t input_alignment; member107 key.input_alignment = cudnn_utils::getAlignment(input); in apply_impl_helper()149 …ls::getTensorDescriptor(input.sizes(), input.strides(), CUDNN_DATA_INT8, 'x', key.input_alignment)) in apply_impl_helper()