Searched refs:CalculateTFStrides (Results 1 – 7 of 7) sorted by relevance
203 auto lhs_strides = CalculateTFStrides(lhs_dims); in CreateMatMulParams()204 auto rhs_strides = CalculateTFStrides(rhs_dims); in CreateMatMulParams()205 auto out_strides = CalculateTFStrides(out_dims); in CreateMatMulParams()
134 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND()137 ReorderStrides(CalculateTFStrides(out_dims), perm); in MKLTransposeND()
420 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()425 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()439 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice()
207 auto strides = CalculateTFStrides(dims); in Compute()
471 auto src_strides = CalculateTFStrides(src_dims); in Compute()627 auto src_strides = CalculateTFStrides(src_dims); in Compute()
680 dst_dims_in_nchw, CalculateTFStrides(dst_dims_in_nchw)); in Compute()
179 memory::dims CalculateTFStrides(const memory::dims& dims_tf_order);469 auto strides = CalculateTFStrides(dims); in GetTfLayout()1179 inline memory::dims CalculateTFStrides(const memory::dims& dims_tf_order) { in CalculateTFStrides() function