Home
last modified time | relevance | path

Searched refs:MklDnnDimsInNCHW (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc410 ? MklDnnDimsInNCHW(begin_dims, input_tf_format) in ComputeMklSlice()
412 size_dims = is_slice2d ? MklDnnDimsInNCHW(size_dims, input_tf_format) in ComputeMklSlice()
Dmkl_concat_op.cc672 dst_dims_in_nchw = MklDnnDimsInNCHW( in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1120 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW() function