Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_slice_op.cc76 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs() local
148 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs() local
392 MklDnnShape input_mkl_shape; in ComputeMklSlice() local
444 const MklDnnShape& input_mkl_shape, in AllocateOutputTensor()
Dmkl_pooling_ops_common.h605 const MklDnnShape& input_mkl_shape, in ConfigureInput()
664 const MklDnnShape& input_mkl_shape) { in SanityCheckInput()
Dmkl_conv_grad_input_ops.cc484 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
Dmkl_conv_grad_filter_ops.cc612 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h949 MklDnnShape input_mkl_shape; in GetTfShape() local