Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.h613 memory::desc input_md = in ConfigureInput() local
623 dnn_data_input->SetUsrMem(input_md, &input_tensor); in ConfigureInput()
628 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_md.data.dims[0]; in ConfigureInput()
629 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = input_md.data.dims[1]; in ConfigureInput()
630 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_md.data.dims[2]; in ConfigureInput()
631 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_md.data.dims[3]; in ConfigureInput()
632 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_md.data.dims[4]; in ConfigureInput()
Dmkl_slice_op.cc400 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice() local
401 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
409 auto input_md = in ComputeMklSlice() local
411 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
Dmkl_requantize_per_channel_op.cc98 memory::desc input_md = memory::desc(dims_mkl_order, MklDnnType<qint32>(), in Compute() local
108 memory::primitive_desc(input_md, cpu_engine_); in Compute()
Dmkl_avgpooling_op.cc490 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute() local
520 if (input_md.data.format != pooling_fwd->GetSrcMemoryFormat()) { in Compute()
521 dnn_data_input.SetUsrMem(input_md, &input_tensor); in Compute()
Dmkl_maxpooling_op.cc548 memory::desc input_md = in Compute() local
597 if (input_md.data.format != pooling_fwd->GetSrcMemoryFormat()) { in Compute()
598 dnn_data_input.SetUsrMem(input_md, &input_tensor); in Compute()