Lines Matching defs:strides
97 const size_t* strides) {
105 strides_[ii] = strides[ii];
107 CHECK_EQ(dnnLayoutCreate_F32(&tfLayout_, dimension, sizes, strides),
248 (SIZES_OFFSET(dims) + dims * sizeof(size_t)) // Location of strides
334 const memory::dims& strides,
544 auto strides = CalculateTFStrides(dims);
545 return CreateBlockedMemDescHelper(dims, strides, data_.T_);
911 inline void GetStridesFromSizes(TensorFormat data_format, size_t* strides,
913 // MKL requires strides in NCHW
915 strides[0] = sizes[2];
916 strides[1] = sizes[0] * sizes[2];
917 strides[2] = 1;
918 strides[3] = sizes[0] * sizes[1] * sizes[2];
920 strides[0] = 1;
921 strides[1] = sizes[0];
922 strides[2] = sizes[0] * sizes[1];
923 strides[3] = sizes[0] * sizes[1] * sizes[2];
1369 /// Function to calculate strides given tensor shape in Tensorflow order
1372 /// innermost dimension. So strides for this tensor would be {4 * 3 * 2,
1376 /// @return memory::dims containing strides for the tensor.
1379 memory::dims strides(dims_tf_order.size());
1381 strides[last_dim_idx] = 1;
1383 strides[d] = strides[d + 1] * dims_tf_order[d + 1];
1385 return strides;
1396 /// @input: strides corresponding to dimensions. One can use utility
1397 /// function such as CalculateTFStrides to compute strides
1400 /// for given dimensions and strides.
1402 const memory::dims& strides,
1404 CHECK_EQ(dim.size(), strides.size());
1418 md.layout_desc.blocking.strides[1][i] = 1;
1419 md.layout_desc.blocking.strides[0][i] = strides[i];
1491 /// @input: strides corresponding to dimensions. One can use utility
1492 /// function such as CalculateTFStrides to compute strides
1495 /// for given dimensions and strides.
1497 const memory::dims& strides) {
1498 return CreateBlockedMemDescHelper(dim, strides, MklDnnType<T>());
1502 /// format. So in addition to accepting dimensions, it also accepts strides.
1507 inline void SetUsrMem(const memory::dims& dim, const memory::dims& strides,
1509 CHECK_EQ(dim.size(), strides.size());
1510 auto blocked_md = MklDnnData<T>::CreateBlockedMemDesc(dim, strides);
1514 inline void SetUsrMem(const memory::dims& dim, const memory::dims& strides,
1517 SetUsrMem(dim, strides, GetTensorBuffer(tensor));
