| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 84 if (!input_shape_0.IsMklTensor() && !input_shape_1.IsMklTensor()) { in Compute()
95 if (input_shape_0.IsMklTensor() && input_shape_1.IsMklTensor()) { in Compute()
196 if (input_shape_0.IsMklTensor() && !input_shape_1.IsMklTensor()) { in Compute()
202 } else if (!input_shape_0.IsMklTensor() && input_shape_1.IsMklTensor()) { in Compute()
|
| D | mkl_lrn_op.cc | 108 if (!src_dnn_shape.IsMklTensor()) { in Compute()
285 if (src_dnn_shape.IsMklTensor()) { in SanityCheckInputs()
354 input_grad_dnn_shape.IsMklTensor() && in Compute()
355 orig_input_dnn_shape.IsMklTensor() && in Compute()
356 orig_output_dnn_shape.IsMklTensor() && in Compute()
481 DCHECK(input_grad_dnn_shape.IsMklTensor() == true); in ConfigureInputGradient()
513 if (input_grad_dnn_shape.IsMklTensor()) { in MklDefaultToEigen()
522 if (orig_input_dnn_shape.IsMklTensor()) { in MklDefaultToEigen()
530 if (orig_output_dnn_shape.IsMklTensor()) { in MklDefaultToEigen()
601 if (in_grads_dnn_shape.IsMklTensor()) { in SanityCheckInputs()
[all …]
|
| D | mkl_dequantize_op.cc | 68 auto src_dims = src_mkl_shape.IsMklTensor() in Compute()
84 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
96 if (src_mkl_shape.IsMklTensor()) { in Compute()
108 if (src_mkl_shape.IsMklTensor()) { in Compute()
|
| D | mkl_softmax_op.cc | 211 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
217 if (src_mkl_shape.IsMklTensor()) { 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_avgpooling_op.cc | 91 dnn_shape_input.IsMklTensor() in Compute()
97 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute()
209 orig_input_mkl_shape.IsMklTensor() in Compute()
217 grad_mkl_shape.IsMklTensor() in Compute()
228 orig_input_mkl_shape.IsMklTensor() in Compute()
235 grad_mkl_shape.IsMklTensor() in Compute()
|
| D | mkl_maxpooling_op.cc | 107 dnn_shape_input.IsMklTensor() in Compute()
120 dnn_shape_input.IsMklTensor() in Compute()
265 orig_input_mkl_shape.IsMklTensor() in Compute()
273 grad_mkl_shape.IsMklTensor() in Compute()
285 orig_input_mkl_shape.IsMklTensor() in Compute()
292 grad_mkl_shape.IsMklTensor() in Compute()
|
| D | mkl_relu_op.cc | 466 if (dnn_shape_src.IsMklTensor()) { in Compute()
498 if (is_src_reordered || dnn_shape_src.IsMklTensor()) { in Compute()
503 if (dnn_shape_src.IsMklTensor()) { in Compute()
625 if (!dnn_shape_src.IsMklTensor() && !dnn_shape_diff_dst.IsMklTensor()) { in Compute()
630 } else if (dnn_shape_src.IsMklTensor() && in Compute()
631 !dnn_shape_diff_dst.IsMklTensor()) { in Compute()
643 } else if (!dnn_shape_src.IsMklTensor() && in Compute()
644 dnn_shape_diff_dst.IsMklTensor()) { in Compute()
672 if (dnn_shape_src.IsMklTensor() || dnn_shape_diff_dst.IsMklTensor()) { in Compute()
673 common_md = dnn_shape_src.IsMklTensor() ? src_md : diff_dst_md; in Compute()
[all …]
|
| D | mkl_slice_op.cc | 79 DCHECK_EQ(begin_mkl_shape.IsMklTensor(), false); in ValidateMklInputs()
80 DCHECK_EQ(size_mkl_shape.IsMklTensor(), false); in ValidateMklInputs()
82 TensorShape input_tf_shape = input_mkl_shape.IsMklTensor() in ValidateMklInputs()
404 if (input_mkl_shape.IsMklTensor()) { in ComputeMklSlice()
446 DCHECK_EQ(input_mkl_shape.IsMklTensor(), output_mkl_shape.IsMklTensor()); in ComputeMklSlice()
480 if (input_mkl_shape.IsMklTensor()) { in AllocateOutputTensor()
|
| D | mkl_aggregate_ops.cc | 46 return src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in GetTensorShape()
78 if (src_mkl_shape.IsMklTensor()) { in FindMKLInputIndex()
122 bool input_in_mkl_format = mkl_shape.IsMklTensor(); in Compute()
190 if (src_mkl_shape.IsMklTensor()) { in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 695 if (dnn_shape_src.IsMklTensor()) { in Compute()
732 if (dnn_shape_src.IsMklTensor()) in Compute()
753 if (dnn_shape_src.IsMklTensor()) { in Compute()
768 dnn_shape_src.IsMklTensor() in Compute()
772 auto src_md = dnn_shape_src.IsMklTensor() in Compute()
852 auto ndims = dnn_shape_src.IsMklTensor() ? dnn_shape_src.GetDimension() in Compute()
1085 if (dnn_shape_diff_dst.IsMklTensor()) { in Compute()
1099 if (dnn_shape_src.IsMklTensor()) { in Compute()
1133 if (dnn_shape_src.IsMklTensor()) { in Compute()
1135 } else if (dnn_shape_diff_dst.IsMklTensor()) { in Compute()
[all …]
|
| D | mkl_matmul_op_fused.cc | 67 OP_REQUIRES(ctx, !weight_mkl_shape.IsMklTensor(), in Compute()
71 auto src_tf_shape = src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in Compute()
165 add_mkl_shape.IsMklTensor() in Compute()
212 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
|
| D | mkl_qmatmul_op.cc | 170 OP_REQUIRES(context, !weight_mkl_shape.IsMklTensor(), in Compute()
181 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
184 auto weight_tf_shape = weight_mkl_shape.IsMklTensor() in Compute()
210 src_mkl_shape.IsMklTensor() in Compute()
217 auto weight_md = weight_mkl_shape.IsMklTensor() in Compute()
|
| D | mkl_reshape_op.cc | 48 CHECK_EQ(mkl_shape_input.IsMklTensor(), true); in SkipReorder()
67 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); in Compute()
|
| D | mkl_quantize_op.cc | 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()
371 src_mkl_shape.IsMklTensor() in Compute()
407 if (src_mkl_shape.IsMklTensor()) { in Compute()
|
| D | mkl_identity_op.cc | 42 if (dnn_shape_input.IsMklTensor()) { in Compute()
|
| D | mkl_concat_op.cc | 482 const TensorShape expected_shape = mkl_input_shapes[0].IsMklTensor() in Compute()
491 s.IsMklTensor() ? s.GetTfShape() : input_tensors[i].shape(); in Compute()
514 if (s.IsMklTensor()) in Compute()
737 if (dnn_shape_dst.IsMklTensor()) in Compute()
769 dst_md = dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() in Compute()
827 if (mkl_input_shapes[i].IsMklTensor()) { in CallEigenVersion()
|
| D | mkl_tfconv_op.h | 74 if (!input_shape.IsMklTensor()) { in ConvertMklToTf()
|
| D | mkl_conv_grad_filter_ops.cc | 441 src_mkl_shape.IsMklTensor() in Compute()
449 diff_dst_mkl_shape.IsMklTensor() in Compute()
562 DCHECK(!diff_filter_mkl_shape.IsMklTensor()); in Compute()
626 CHECK(!filter_mkl_shape.IsMklTensor()) in ValidateMklShapes()
|
| D | mkl_conv_grad_input_ops.cc | 400 filter_mkl_shape.IsMklTensor() in Compute()
412 diff_dst_mkl_shape.IsMklTensor() in Compute()
517 CHECK(!input_mkl_shape.IsMklTensor()) in ValidateMklShapes()
|
| D | mkl_pooling_ops_common.h | 502 if (!original_input_mkl_shape.IsMklTensor()) { in InitMklPoolParameters()
616 input_mkl_shape.IsMklTensor() in ConfigureInput()
671 if (!input_mkl_shape.IsMklTensor()) { in SanityCheckInput()
|
| D | mkl_conv_ops.cc | 507 OP_REQUIRES(context, !filter_mkl_shape.IsMklTensor(), in Compute()
612 src_mkl_shape.IsMklTensor() in Compute()
624 DCHECK(!filter_mkl_shape.IsMklTensor()); in Compute()
626 filter_mkl_shape.IsMklTensor() in Compute()
903 add_mkl_shape.IsMklTensor() in AllocateOutputTensor()
1817 summand_mkl_shape.IsMklTensor() in AllocateOutputTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 285 if (this->IsMklTensor() != input_shape.IsMklTensor()) {
291 if (this->IsMklTensor()) {
306 if (!this->IsMklTensor()) {
313 inline const bool IsMklTensor() const { return data_.is_mkl_tensor_; } in IsMklTensor() function
483 return IsMklTensor() ? GetMklLayout() : GetTfLayout(); in GetCurLayout()
628 if (!input_mkl_shape.IsMklTensor()) { in ConvertMklToTF()
731 if (input_mkl_shape.IsMklTensor() && !eager_mode) {
|