Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_requantize_per_channel_op.cc98 memory::desc input_md = memory::desc(dims_mkl_order, MklDnnType<qint32>(), in Compute() local
Dmkl_slice_op.cc400 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice() local
409 auto input_md = in ComputeMklSlice() local
Dmkl_avgpooling_op.cc490 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute() local
Dmkl_maxpooling_op.cc548 memory::desc input_md = in Compute() local
Dmkl_pooling_ops_common.h613 memory::desc input_md = in ConfigureInput() local