| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_softmax_op.cc | 78 auto src_md = variable
89 src.SetUsrMem(src_md, &src_tensor);
124 dst.SetUsrMem(src_md, output_tensor);
|
| mkl_relu_op.cc | 402 memory::desc src_md({}, memory::data_undef, memory::format_undef);
404 src_md = dnn_shape_src.GetMklLayout();
409 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides);
411 src.SetUsrMem(src_md, &src_tensor);
442 auto dst_md = src_md;
498 memory::desc src_md({}, memory::data_undef, memory::format_undef);
515 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides);
516 diff_dst_md = src_md;
523 src_md = dnn_shape_src.GetMklLayout();
543 src_md
[all...] |
| mkl_fused_batch_norm_op.cc | 813 auto src_md = dnn_shape_src.IsMklTensor() variable
[all...] |
| mkl_concat_op.cc | 697 auto src_md = local
704 srcs[k].SetUsrMem(src_md, &input_tensors[k]);
[all...] |
| mkl_conv_ops.cc | 574 auto src_md = src_mkl_shape.IsMklTensor() variable
577 src.SetUsrMem(src_md, &src_tensor);
[all...] |
| mkl_lrn_op.cc | 796 memory::desc src_md = src_dnn_shape.GetCurLayout();
803 src_dnn_data.SetUsrMem(src_md, &src_tensor);
807 dst_dnn_data.SetUsrMem(src_md);
[all...] |
