Searched refs:s_dims (Results 1 – 2 of 2) sorted by relevance
492 size_t s_dims = s_shape.dims(); in Compute() local495 context, s_dims == expected_dims, in Compute()499 s_dims, " vs. expected rank = ", expected_dims)); in Compute()519 if (s_dims != 4 && s_dims != 2) invoke_eigen = true; in Compute()640 size_t s_dims = s_shape.dims(); in Compute() local644 if (s_dims == 4) in Compute()646 else if (s_dims == 2) in Compute()
211 memory::dims s_dims = {2, fwdParams.depth}; in Setup() local215 new memory({{s_dims}, MklDnnType<U>(), memory::format_tag::nc}, in Setup()