Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_slice_op.cc421 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice() local
434 auto input_md = in ComputeMklSlice() local
H A Dmkl_requantize_per_channel_op.cc113 memory::desc input_md = memory::desc(dims_mkl_order, MklDnnType<qint32>(), in Compute() local
H A Dmkl_avgpooling_op.cc95 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute() local
H A Dmkl_maxpooling_op.cc101 memory::desc input_md = in Compute() local
H A Dmkl_pooling_ops_common.h630 memory::desc input_md = in ConfigureInput() local