| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_relu_op.cc | 45 memory::desc src_md; member in tensorflow::MklEltwiseFwdParams
50 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams() argument
53 src_md(src_md), in MklEltwiseFwdParams()
111 std::shared_ptr<memory::desc> src_md; member
129 src_md(nullptr), in EltwiseFwdContext()
138 context_.src_md.reset(new memory::desc(fwdParams.src_md.data)); in Setup()
139 context_.src_mpd.reset(new memory::desc(*context_.src_md)); in Setup()
143 prop_kind::forward, fwdParams.alg_kind, *context_.src_md, in Setup()
297 std::shared_ptr<memory::desc> src_md; member
323 src_md(nullptr), in EltwiseBwdContext()
[all …]
|
| D | mkl_quantize_op.cc | 61 memory::desc src_md; member
70 MklReorderWithScaleFwdParams(memory::dims src_dims, memory::desc src_md, in MklReorderWithScaleFwdParams()
72 : src_dims(src_dims), src_md(src_md), dst_md(dst_md) {} in MklReorderWithScaleFwdParams()
130 new memory(fwdParams.src_md, cpu_engine_, DummyData)); in Setup()
370 auto src_md = in Compute() local
396 src.SetUsrMem(src_md, &min_shifted_input_tensor); in Compute()
398 src.SetUsrMem(src_md, &src_tensor); in Compute()
468 MklReorderWithScaleFwdParams fwdParams(src_dims, src_md, dst_md); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 49 memory::desc src_md; member
52 bool training, memory::desc src_md, in MklBatchNormFwdParams()
59 src_md(src_md) {} in MklBatchNormFwdParams()
197 auto src_md = fwdParams.src_md; in Setup() local
200 context_.pkind, src_md, fwdParams.eps, in Setup()
384 memory::desc src_md; member
389 memory::desc src_md, memory::desc diff_dst_md) in MklBatchNormBwdParams()
395 src_md(src_md), in MklBatchNormBwdParams()
526 auto src_md = bwdParams.src_md; in Setup() local
543 prop_kind::forward_training, src_md, bwdParams.eps, bn_flags); in Setup()
[all …]
|
| D | mkl_pooling_ops_common.cc | 46 context_.src_md.reset(new memory::desc(fwdParams.src_md.data)); in Setup()
54 fwdParams.prop_kind, fwdParams.alg_kind, *context_.src_md, in Setup()
138 context_.src_md.reset(new memory::desc({bwdParams.src_dims}, MklDnnType<T>(), in Setup()
140 context_.src_md.reset(new memory::desc(bwdParams.src_md.data)); in Setup()
150 bwdParams.alg_kind, *context_.src_md, *context_.dst_md, bwdParams.strides, in Setup()
155 bwdParams.prop_kind, bwdParams.alg_kind, *context_.src_md, in Setup()
|
| D | mkl_concat_op.cc | 324 std::vector<mkldnn::memory::desc> src_md; member
346 context_.src_md.push_back(source_md); in Setup()
358 *context_.dst_md, concat_fwd_dims.concat_dims, context_.src_md, in Setup()
610 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
611 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
621 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
622 srcs[k].SetUsrMem(src_md, &input_tensors[k]); in Compute()
626 memory::dims src_dims(src_md.data.dims, in Compute()
627 &src_md.data.dims[src_md.data.ndims]); in Compute()
628 src_md = in Compute()
[all …]
|
| D | mkl_softmax_op.cc | 95 std::shared_ptr<memory::desc> src_md; member
108 src_md(nullptr), in SoftmaxFwdContext()
117 context_.src_md.reset( in Setup()
122 prop_kind::forward_scoring, *context_.src_md, fwdParams.axis)); in Setup()
128 new memory(*context_.src_md, cpu_engine_, DummyData)); in Setup()
|
| D | mkl_pooling_ops_common.h | 50 memory::desc src_md; member
57 memory::format_tag src_format, memory::desc src_md, in MklPoolingParams()
68 src_md(src_md), in MklPoolingParams()
126 std::shared_ptr<mkldnn::memory::desc> src_md; member
145 src_md(nullptr), in PoolingFwdContext()
263 std::shared_ptr<mkldnn::memory::desc> src_md; member
286 src_md(nullptr), in PoolingBwdContext()
|
| D | mkl_dequantize_op.cc | 84 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
89 src.SetUsrMem(src_md, &src_tensor); in Compute()
|
| D | mkl_matmul_op_fused.cc | 212 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
216 if (src_md != matmul_pd->src_desc()) { in Compute()
217 src_mkl.SetUsrMem(src_md, src_data); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 180 std::shared_ptr<mkldnn::memory::desc> src_md; member
198 src_md(nullptr), in ConvBwdFilterContext()
215 context_.src_md.reset(new memory::desc({convBwdFilterDims.src_dims}, in Setup()
233 *context_.src_md, *context_.diff_filter_md, *context_.diff_dst_md, in Setup()
241 mkldnn::algorithm::convolution_direct, *context_.src_md, in Setup()
248 mkldnn::algorithm::convolution_direct, *context_.src_md, in Setup()
|
| D | mkl_qmatmul_op.cc | 209 auto src_md = in Compute() local
213 src.SetUsrMem(src_md, &src_tensor); in Compute()
249 if (src_md != matmul_fwd_pd->src_desc()) { in Compute()
250 src.SetUsrMem(src_md, &src_tensor); in Compute()
|
| D | mkl_avgpooling_op.cc | 227 memory::desc src_md = in Compute() local
247 static_cast<memory::format_tag>(this->data_format_mkldnn_), src_md, in Compute()
|
| D | mkl_matmul_ops_common.h | 151 std::shared_ptr<mkldnn::memory::desc> src_md; member
169 src_md(nullptr), in MklDnnMatMulFwdContext()
179 context_.src_md.reset(new memory::desc({matmul_fwd_params.src_dims}, in Setup()
196 prop_kind::forward_inference, *context_.src_md, *context_.weight_md, in Setup()
|
| D | mkl_maxpooling_op.cc | 284 memory::desc src_md = in Compute() local
301 static_cast<memory::format_tag>(this->data_format_mkldnn_), src_md, in Compute()
|
| D | mkl_conv_ops.cc | 184 std::shared_ptr<mkldnn::memory::desc> src_md; member
202 src_md(nullptr), in ConvFwdContext()
217 context_.src_md.reset(new memory::desc( in Setup()
236 *context_.src_md, *context_.filter_md, *context_.bias_md, in Setup()
242 *context_.src_md, *context_.filter_md, *context_.dst_md, in Setup()
611 auto src_md = in Compute() local
615 src.SetUsrMem(src_md, &src_tensor); in Compute()
681 if (src_md != conv_fwd_pd->src_desc()) { in Compute()
682 src.SetUsrMem(src_md, &src_tensor); in Compute()
|
| D | mkl_lrn_op.cc | 130 memory::desc src_md = src_dnn_shape.GetCurLayout(); in Compute() local
137 src_dnn_data.SetUsrMem(src_md, &src_tensor); in Compute()
142 dst_dnn_data.SetUsrMem(src_md); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 64 #define IS_SRC_REORDER_NEEDED(src_md, op_pd, op) src_md != op_pd->src_desc() argument
111 #define REORDER_PD_CONSTRUCTOR(src_md, dst_md, engine) \ argument
112 ReorderPd(engine, src_md, engine, dst_md)
113 #define REORDER_PD_CONSTRUCTOR_WITH_ATTR(src_md, dst_md, engine, prim_attr) \ argument
114 ReorderPd(engine, src_md, engine, dst_md, prim_attr)
|