Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_requantize_per_channel_op.cc103 memory::desc input_md = memory::desc(dims_mkl_order, MklDnnType<qint32>(), in Compute() local
Dmkl_slice_op.cc414 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice() local
427 auto input_md = in ComputeMklSlice() local
Dmkl_avgpooling_op.cc97 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute() local
Dmkl_maxpooling_op.cc106 memory::desc input_md = in Compute() local
Dmkl_pooling_ops_common.h615 memory::desc input_md = in ConfigureInput() local
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h115 #define SKIP_INPUT_REORDER(input_mkl_shape, input_md) \ argument