| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_relu_op.cc | 44 memory::desc src_md; member in class:tensorflow::MklEltwiseFwdParams
49 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md,
52 src_md(src_md),
65 static_cast<mkldnn::memory::format>(fwdParams.src_md.data.format);
111 std::shared_ptr<memory::desc> src_md; member in struct:tensorflow::MklEltwiseFwdPrimitive::EltwiseFwdContext
129 src_md(nullptr),
139 context_.src_md.reset(new memory::desc(fwdParams.src_md.data));
141 new memory::primitive_desc(*context_.src_md, cpu_engine_))
302 std::shared_ptr<memory::desc> src_md; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
[all...] |
| mkl_concat_op.cc | 305 auto src_md = mkl_input_shapes[k].GetMklLayout(); local
306 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
316 auto src_md = mkl_input_shapes[k].GetMklLayout(); local
317 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
319 if (src_md.data.format != mkl_common_format) {
320 memory::dims src_dims(src_md.data.dims,
321 &src_md.data.dims[src_md.data.ndims]);
322 src_md =
326 srcs_pd.push_back(memory::primitive_desc(src_md, cpu_engine))
338 auto src_md = local
[all...] |
| mkl_softmax_op.cc | 121 auto src_md = src_mkl_shape.IsMklTensor() variable
127 src.SetUsrMem(src_md, &src_tensor);
160 dst.SetUsrMem(src_md, output_tensor);
|
| mkl_fused_batch_norm_op.cc | 147 auto src_md = memory::desc({fwdParams.src_dims}, MklDnnType<T>(), local
152 context_.pkind, src_md, fwdParams.eps, context_.flags);
157 context_.src_mem.reset(new memory({src_md, cpu_engine_}, DummyData));
388 auto src_md = memory::desc({bwdParams.src_dims}, MklDnnType<T>(), local
403 prop_kind::forward_training, src_md, bwdParams.eps,
416 prop_kind::backward, diff_dst_md, src_md, bwdParams.eps,
423 context_.src_mem.reset(new memory({src_md, cpu_engine_}, DummyData));
433 context_.diff_src_mem.reset(new memory({src_md, cpu_engine_}, DummyData));
615 auto src_md = dnn_shape_src.IsMklTensor() variable
646 if (src_md.data.format != bn_fwd->GetSrcFmt())
935 memory::desc src_md = variable
[all...] |
| mkl_conv_grad_filter_ops.cc | 180 std::shared_ptr<mkldnn::memory::desc> src_md; member in struct:tensorflow::MklConvBwdFilterPrimitive::ConvBwdFilterContext
200 src_md(nullptr),
210 context_.src_md.reset(new memory::desc(
229 convolution_direct, *context_.src_md, *context_.diff_filter_md,
236 convolution_direct, *context_.src_md, *context_.diff_filter_md,
244 prop_kind::forward, convolution_direct, *context_.src_md,
|
| mkl_conv_ops.cc | 186 std::shared_ptr<mkldnn::memory::desc> src_md; member in struct:tensorflow::MklConvFwdPrimitive::ConvFwdContext
206 src_md(nullptr),
216 context_.src_md.reset(new memory::desc(
232 prop_kind::forward, convolution_direct, *context_.src_md,
238 prop_kind::forward, convolution_direct, *context_.src_md,
1030 auto src_md = src_mkl_shape.IsMklTensor() variable
[all...] |
| mkl_pooling_ops_common.cc | 57 context_.src_md.reset(new memory::desc({fwdParams.src_dims}, MklDnnType<T>(),
64 fwdParams.prop_kind, fwdParams.alg_kind, *context_.src_md,
|
| mkl_lrn_op.cc | 126 memory::desc src_md = src_dnn_shape.GetCurLayout(); variable
133 src_dnn_data.SetUsrMem(src_md, &src_tensor);
137 dst_dnn_data.SetUsrMem(src_md);
|
| mkl_pooling_ops_common.h | 125 std::shared_ptr<mkldnn::memory::desc> src_md; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
142 src_md(nullptr),
|
