Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
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()
Dmkl_pooling_ops_common.h622 const MklDnnShape& input_mkl_shape, in ConfigureInput()
685 const MklDnnShape& input_mkl_shape) { in SanityCheckInput()
Dmkl_conv_grad_input_ops.cc514 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
Dmkl_conv_grad_filter_ops.cc652 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h628 const MklDnnShape& input_mkl_shape, in ConvertMklToTF()
739 MklDnnShape input_mkl_shape; variable