Searched refs:conv1_src_dim (Results 1 – 1 of 1) sorted by relevance
68 memory::dims conv1_src_dim = {ToInt(input_batch), ToInt(input_channels), in MKLConvImpl() local87 memory({{{conv1_src_dim}, memory::data_type::f32, memory::format::nhwc}, in MKLConvImpl()102 {conv1_src_dim}, memory::data_type::f32, memory::format::any); in MKLConvImpl()