| /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); in Compute()
68 auto src_dims = src_mkl_shape.IsMklTensor() in Compute()
69 ? src_mkl_shape.GetSizesAsMklDnnDims() in Compute()
84 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
85 ? src_mkl_shape.GetMklLayout() in Compute()
96 if (src_mkl_shape.IsMklTensor()) { in Compute()
97 dst_md = memory::desc(src_mkl_shape.GetMklLayout().data); in Compute()
108 if (src_mkl_shape.IsMklTensor()) { in Compute()
112 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()
184 MklDnnShape src_mkl_shape; in Compute() local
185 GetMklShape(ctx, src_idx, &src_mkl_shape); in Compute()
190 if (src_mkl_shape.IsMklTensor()) { in Compute()
191 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); 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); in Compute()
181 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
182 ? src_mkl_shape.GetTfShape() in Compute()
210 src_mkl_shape.IsMklTensor() in Compute()
211 ? 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()
441 src_mkl_shape.IsMklTensor() in Compute()
442 ? 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 | 502 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute() local
503 GetMklShape(context, kInputIndex_Src, &src_mkl_shape, native_format); in Compute()
612 src_mkl_shape.IsMklTensor() in Compute()
613 ? src_mkl_shape.GetMklLayout() in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 39 #define GET_FORMAT_FROM_SHAPE(src_mkl_shape) MklTensorFormat::FORMAT_BLOCKED argument
|