Home
last modified time | relevance | path

Searched refs:GetMklLayout (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_input_conversion_op.cc91 auto input0_md = input_shape_0.GetMklLayout(); in Compute()
92 auto input1_md = input_shape_1.GetMklLayout(); in Compute()
219 auto output_mkl_md = mkl_shape->GetMklLayout(); in Compute()
Dmkl_avgpooling_op.cc96 ? dnn_shape_input.GetMklLayout() in Compute()
239 ? orig_input_mkl_shape.GetMklLayout() in Compute()
246 ? grad_mkl_shape.GetMklLayout() in Compute()
Dmkl_maxpooling_op.cc103 ? dnn_shape_input.GetMklLayout() in Compute()
292 ? orig_input_mkl_shape.GetMklLayout() in Compute()
299 ? grad_mkl_shape.GetMklLayout() in Compute()
Dmkl_tfconv_op.h92 auto input_mkl_md = input_shape.GetMklLayout(); in ConvertMklToTf()
Dmkl_matmul_op_fused.cc180 ? add_mkl_shape.GetMklLayout() in Compute()
227 ? src_mkl_shape.GetMklLayout() in Compute()
Dmkl_concat_op.cc622 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute()
633 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute()
751 dst_md = dnn_shape_dst.GetMklLayout(); in Compute()
782 dst_md = dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() in Compute()
Dmkl_relu_op.cc468 src_md = dnn_shape_src.GetMklLayout(); in Compute()
634 src_md = dnn_shape_src.GetMklLayout(); in Compute()
647 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
663 src_md = dnn_shape_src.GetMklLayout(); in Compute()
664 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
Dmkl_reshape_op.cc152 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in Compute()
Dmkl_qmatmul_op.cc212 ? src_mkl_shape.GetMklLayout() in Compute()
219 ? weight_mkl_shape.GetMklLayout() in Compute()
Dmkl_aggregate_ops.cc191 md = src_mkl_shape.GetMklLayout(); in Compute()
Dmkl_conv_grad_input_ops.cc402 ? filter_mkl_shape.GetMklLayout() in Compute()
414 ? diff_dst_mkl_shape.GetMklLayout() in Compute()
Dmkl_conv_grad_filter_ops.cc446 ? src_mkl_shape.GetMklLayout() in Compute()
454 ? diff_dst_mkl_shape.GetMklLayout() in Compute()
Dmkl_conv_ops.cc750 ? src_mkl_shape.GetMklLayout() in Compute()
765 ? filter_mkl_shape.GetMklLayout() in Compute()
1140 ? add_mkl_shape.GetMklLayout() in AllocateOutputTensor()
2027 auto dst_md = summand_mkl_shape.GetMklLayout(); in AllocateOutputTensor()
2095 ? summand_mkl_shape.GetMklLayout() in AllocateOutputTensor()
Dmkl_quantize_op.cc381 ? src_mkl_shape.GetMklLayout() in Compute()
Dmkl_fused_batch_norm_op.cc834 ? dnn_shape_src.GetMklLayout() in Compute()
1265 ? dnn_shape_src.GetMklLayout() in Compute()
1269 ? dnn_shape_diff_dst.GetMklLayout() in Compute()
Dmkl_slice_op.cc421 auto input_md = input_mkl_shape.GetMklLayout(); in ComputeMklSlice()
Dmkl_pooling_ops_common.h632 ? input_mkl_shape.GetMklLayout() in ConfigureInput()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h290 const dnnl_memory_desc_t& cur_md = (this->GetMklLayout()).data;
292 input_shape.GetMklLayout().data;
428 inline const memory::desc GetMklLayout() const { in GetMklLayout() function
480 return IsMklTensor() ? GetMklLayout() : GetTfLayout(); in GetCurLayout()
646 auto input_mkl_md = input_mkl_shape.GetMklLayout(); in ConvertMklToTF()