Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc73 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs() local
146 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs() local
401 MklDnnShape input_mkl_shape; in ComputeMklSlice() local
470 const MklDnnShape& input_mkl_shape, in AllocateOutputTensor()
Dmkl_pooling_ops_common.h607 const MklDnnShape& input_mkl_shape, in ConfigureInput()
670 const MklDnnShape& input_mkl_shape) { in SanityCheckInput()
Dmkl_conv_grad_input_ops.cc511 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
Dmkl_conv_grad_filter_ops.cc623 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h115 #define SKIP_INPUT_REORDER(input_mkl_shape, input_md) \ argument
Dmkl_util.h625 const MklDnnShape& input_mkl_shape, in ConvertMklToTF()
729 MklDnnShape input_mkl_shape; variable