Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc408 bool is_slice2d = (input_mkl_shape.GetDimension() == 4); in ComputeMklSlice() local
409 begin_dims = is_slice2d in ComputeMklSlice()
412 size_dims = is_slice2d ? MklDnnDimsInNCHW(size_dims, input_tf_format) in ComputeMklSlice()