Searched refs:input_md (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.h | 615 memory::desc input_md = in ConfigureInput() local
625 dnn_data_input->SetUsrMem(input_md, &input_tensor); in ConfigureInput()
630 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_md.data.dims[0]; in ConfigureInput()
631 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = input_md.data.dims[1]; in ConfigureInput()
632 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_md.data.dims[2]; in ConfigureInput()
633 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_md.data.dims[3]; in ConfigureInput()
634 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_md.data.dims[4]; in ConfigureInput()
|
| D | mkl_requantize_per_channel_op.cc | 103 memory::desc input_md = memory::desc(dims_mkl_order, MklDnnType<qint32>(), in Compute() local
124 new memory(input_md, cpu_engine_, input_buf)); in Compute()
|
| D | mkl_slice_op.cc | 414 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice() local
415 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
427 auto input_md = in ComputeMklSlice() local
429 src.SetUsrMem(input_md, &input_tensor); in ComputeMklSlice()
|
| D | mkl_avgpooling_op.cc | 97 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute() local
117 static_cast<memory::format_tag>(this->data_format_mkldnn_), input_md, in Compute()
|
| D | mkl_maxpooling_op.cc | 106 memory::desc input_md = in Compute() local
142 static_cast<memory::format_tag>(this->data_format_mkldnn_), input_md, in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 115 #define SKIP_INPUT_REORDER(input_mkl_shape, input_md) \ argument
|