Home
last modified time | relevance | path

Searched refs:src_mkl_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_softmax_op.cc54 MklDnnShape src_mkl_shape; in Compute() local
55 GetMklShape(context, src_idx, &src_mkl_shape); in Compute()
59 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
60 ? src_mkl_shape.GetTfShape() in Compute()
66 if (src_mkl_shape.IsMklTensor()) { in Compute()
68 output_dims = src_mkl_shape.GetSizesAsMklDnnDims(); in Compute()
94 if (src_mkl_shape.IsMklTensor()) { in Compute()
101 if (src_mkl_shape.IsMklTensor()) { in Compute()
121 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
122 ? src_mkl_shape.GetMklLayout() in Compute()
[all …]
Dmkl_conv_grad_filter_ops.cc384 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
385 GetMklShape(context, kInputIdx, &src_mkl_shape); in Compute()
389 ValidateMklShapes(src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape); in Compute()
443 src_mkl_shape.IsMklTensor() in Compute()
444 ? src_mkl_shape.GetMklLayout() in Compute()
Dmkl_conv_grad_input_ops.cc321 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
322 GetMklShape(context, kInputIdx, &src_mkl_shape); in Compute()
326 ValidateMklShapes(src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape); in Compute()
Dmkl_conv_ops.cc930 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute() local
931 GetMklShape(context, kInputIndex_Src, &src_mkl_shape); in Compute()
1030 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
1031 ? src_mkl_shape.GetMklLayout() in Compute()