| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_dequantize_op.cc | 63 MklDnnShape src_mkl_shape; in Compute() local
64 GetMklShape(ctx, kSrcIndex, &src_mkl_shape, native_format); in Compute()
68 auto src_dims = src_mkl_shape.IsMklTensor() in Compute()
69 ? src_mkl_shape.GetSizesAsMklDnnDims() in Compute()
86 src_mkl_shape.IsMklTensor() in Compute()
87 ? src_mkl_shape.GetMklLayout() in Compute()
99 if (src_mkl_shape.IsMklTensor()) { in Compute()
100 dst_md = memory::desc(src_mkl_shape.GetMklLayout().data); in Compute()
112 if (src_mkl_shape.IsMklTensor()) { in Compute()
116 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
[all …]
|
| D | mkl_aggregate_ops.cc | 44 MklDnnShape src_mkl_shape; in GetTensorShape() local
45 GetMklShape(ctx, src_index, &src_mkl_shape); in GetTensorShape()
46 return src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in GetTensorShape()
75 MklDnnShape src_mkl_shape; in FindMKLInputIndex() local
77 GetMklShape(ctx, i, &src_mkl_shape); in FindMKLInputIndex()
78 if (src_mkl_shape.IsMklTensor()) { in FindMKLInputIndex()
185 MklDnnShape src_mkl_shape; in Compute() local
186 GetMklShape(ctx, src_idx, &src_mkl_shape); in Compute()
191 if (src_mkl_shape.IsMklTensor()) { in Compute()
192 md = src_mkl_shape.GetMklLayout(); in Compute()
|
| D | mkl_softmax_op.cc | 206 MklDnnShape src_mkl_shape; in Compute() local
207 GetMklShape(context, src_idx, &src_mkl_shape); in Compute()
211 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
212 ? src_mkl_shape.GetTfShape() in Compute()
217 if (src_mkl_shape.IsMklTensor()) { in Compute()
218 src_dims = src_mkl_shape.GetSizesAsMklDnnDims(); in Compute()
245 if (src_mkl_shape.IsMklTensor()) { in Compute()
252 if (src_mkl_shape.IsMklTensor()) { in Compute()
266 auto src_fmt = src_mkl_shape.IsMklTensor() in Compute()
284 if (src_mkl_shape.IsMklTensor()) { in Compute()
|
| D | mkl_quantize_op.cc | 332 MklDnnShape src_mkl_shape; in Compute() local
333 GetMklShape(ctx, src_idx, &src_mkl_shape, native_format); in Compute()
334 auto src_tf_shape = src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in Compute()
336 auto src_dims = src_mkl_shape.IsMklTensor() in Compute()
337 ? src_mkl_shape.GetSizesAsMklDnnDims() in Compute()
371 src_mkl_shape.IsMklTensor() in Compute()
372 ? src_mkl_shape.GetMklLayout() in Compute()
407 if (src_mkl_shape.IsMklTensor()) { in Compute()
411 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
412 src_mkl_shape.GetSizesAsMklDnnDims(), in Compute()
[all …]
|
| D | mkl_matmul_op_fused.cc | 63 MklDnnShape src_mkl_shape; in Compute() local
65 GetMklShape(ctx, this->kInputIndexSrc, &src_mkl_shape, native_format); in Compute()
71 auto src_tf_shape = src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in Compute()
212 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
213 ? src_mkl_shape.GetMklLayout() in Compute()
|
| D | mkl_qmatmul_op.cc | 167 MklDnnShape src_mkl_shape, weight_mkl_shape; in Compute() local
168 GetMklShape(context, this->kInputIndexSrc, &src_mkl_shape, native_format); in Compute()
182 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
183 ? src_mkl_shape.GetTfShape() in Compute()
211 src_mkl_shape.IsMklTensor() in Compute()
212 ? src_mkl_shape.GetMklLayout() in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 376 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
377 GetMklShape(context, kInputIdx, &src_mkl_shape, native_format); in Compute()
381 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()
|
| D | mkl_conv_grad_input_ops.cc | 319 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute() local
320 GetMklShape(context, kInputIdx, &src_mkl_shape, native_format); in Compute()
324 ValidateMklShapes(src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape); in Compute()
|
| D | mkl_conv_ops.cc | 615 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute() local
616 GetMklShape(context, kInputIndex_Src, &src_mkl_shape, native_format); in Compute()
727 src_mkl_shape.IsMklTensor() in Compute()
728 ? src_mkl_shape.GetMklLayout() in Compute()
|