Searched refs:dims_tf_order (Results 1 – 1 of 1) sorted by relevance
418 memory::dims CalculateTFStrides(const memory::dims& dims_tf_order);1590 inline memory::dims CalculateTFStrides(const memory::dims& dims_tf_order) { in CalculateTFStrides() argument1591 CHECK_GT(dims_tf_order.size(), 0); in CalculateTFStrides()1592 memory::dims strides(dims_tf_order.size()); in CalculateTFStrides()1593 int last_dim_idx = dims_tf_order.size() - 1; in CalculateTFStrides()1596 strides[d] = strides[d + 1] * dims_tf_order[d + 1]; in CalculateTFStrides()