Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_slice_op.cc398 begin_dims = MklDnnDimsInNCHW(begin_dims, input_tf_format); in ComputeMklSlice()
399 size_dims = MklDnnDimsInNCHW(size_dims, input_tf_format); in ComputeMklSlice()
Dmkl_concat_op.cc355 dst_dims_in_nchw = MklDnnDimsInNCHW( in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1549 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW() function