Home
last modified time | relevance | path

Searched defs:input_mkl_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_slice_op.cc76 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs() local
149 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs() local
408 MklDnnShape input_mkl_shape; in ComputeMklSlice() local
478 const MklDnnShape& input_mkl_shape, in AllocateOutputTensor()
H A Dmkl_pooling_ops_common.h622 const MklDnnShape& input_mkl_shape, in ConfigureInput()
685 const MklDnnShape& input_mkl_shape) { in SanityCheckInput()
H A Dmkl_conv_grad_input_ops.cc514 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
H A Dmkl_conv_grad_filter_ops.cc652 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h628 const MklDnnShape& input_mkl_shape, in ConvertMklToTF()
739 MklDnnShape input_mkl_shape; variable