Home
last modified time | relevance | path

Searched refs:CalculateTFStrides (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_transpose_op.cc133 memory::dims in_strides = CalculateTFStrides(in_dims); in MKLTransposeND()
136 ReorderStrides(CalculateTFStrides(out_dims), perm); in MKLTransposeND()
Dmkl_slice_op.cc407 input_strides = CalculateTFStrides(input_dims); in ComputeMklSlice()
415 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice()
Dmkl_reshape_op.cc179 auto output_strides = CalculateTFStrides(output_dims); in Compute()
Dmkl_aggregate_ops.cc128 strides = CalculateTFStrides(dims); in Compute()
Dmkl_relu_op.cc478 auto src_strides = CalculateTFStrides(src_dims); in Compute()
591 auto src_strides = CalculateTFStrides(src_dims); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h418 memory::dims CalculateTFStrides(const memory::dims& dims_tf_order);
655 auto strides = CalculateTFStrides(dims); in GetTfLayout()
1590 inline memory::dims CalculateTFStrides(const memory::dims& dims_tf_order) { in CalculateTFStrides() function