Searched refs:input_tf_format (Results 1 – 1 of 1) sorted by relevance
406 auto input_tf_format = MklDnnDataFormatToTFDataFormat(input_mkl_format); in ComputeMklSlice() local410 ? MklDnnDimsInNCHW(begin_dims, input_tf_format) in ComputeMklSlice()411 : MklDnnDimsInNCDHW(begin_dims, input_tf_format); in ComputeMklSlice()412 size_dims = is_slice2d ? MklDnnDimsInNCHW(size_dims, input_tf_format) in ComputeMklSlice()413 : MklDnnDimsInNCDHW(size_dims, input_tf_format); in ComputeMklSlice()