| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 86 if (!input_shape_0.IsMklTensor() && !input_shape_1.IsMklTensor()) { in Compute()
97 if (input_shape_0.IsMklTensor() && input_shape_1.IsMklTensor()) { in Compute()
194 if (input_shape_0.IsMklTensor() && !input_shape_1.IsMklTensor()) { in Compute()
200 } else if (!input_shape_0.IsMklTensor() && input_shape_1.IsMklTensor()) { in Compute()
|
| D | mkl_softmax_op.cc | 59 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute()
66 if (src_mkl_shape.IsMklTensor()) { 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()
144 if (src_mkl_shape.IsMklTensor()) { in Compute()
|
| D | mkl_maxpooling_op.cc | 81 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
155 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives()
182 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup()
244 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
320 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayouts()
321 bool outbackprop_in_mkl_format = output_backprop_shape.IsMklTensor(); in MklCreateLayouts()
393 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklPrepareInputs()
394 bool outbackprop_in_mkl_format = output_backprop_shape.IsMklTensor(); in MklPrepareInputs()
472 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup()
473 bool outbackprop_in_mkl_format = output_backprop_shape.IsMklTensor(); in MklCleanup()
[all …]
|
| D | mkl_avgpooling_op.cc | 72 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
163 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives()
201 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup()
249 mkl_context.out_backprop_shape.IsMklTensor(); in Compute()
355 bool outbackprop_in_mkl_format = out_backprop_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives()
414 bool outbackprop_in_mkl_format = out_backprop_shape.IsMklTensor(); in MklCleanup()
484 dnn_shape_input.IsMklTensor() in Compute()
490 memory::desc input_md = dnn_shape_input.IsMklTensor() in Compute()
604 orig_input_mkl_shape.IsMklTensor() in Compute()
612 grad_mkl_shape.IsMklTensor() in Compute()
[all …]
|
| D | mkl_lrn_op.cc | 106 if (!src_dnn_shape.IsMklTensor()) { in Compute()
292 if (src_dnn_shape.IsMklTensor()) { in SanityCheckInputs()
358 input_grad_dnn_shape.IsMklTensor() && in Compute()
359 orig_input_dnn_shape.IsMklTensor() && in Compute()
360 orig_output_dnn_shape.IsMklTensor() && in Compute()
474 CHECK_EQ(input_grad_dnn_shape.IsMklTensor(), true); in ConfigureInputGradient()
533 if (input_grad_dnn_shape.IsMklTensor()) { in MklDefaultToEigen()
540 if (orig_input_dnn_shape.IsMklTensor()) { in MklDefaultToEigen()
547 if (orig_output_dnn_shape.IsMklTensor()) { in MklDefaultToEigen()
617 if (in_grads_dnn_shape.IsMklTensor()) { in SanityCheckInputs()
[all …]
|
| D | mkl_slice_op.cc | 82 DCHECK_EQ(begin_mkl_shape.IsMklTensor(), false); in ValidateMklInputs()
83 DCHECK_EQ(size_mkl_shape.IsMklTensor(), false); in ValidateMklInputs()
85 TensorShape input_tf_shape = input_mkl_shape.IsMklTensor() in ValidateMklInputs()
395 if (input_mkl_shape.IsMklTensor()) { in ComputeMklSlice()
422 DCHECK_EQ(input_mkl_shape.IsMklTensor(), output_mkl_shape.IsMklTensor()); in ComputeMklSlice()
454 if (input_mkl_shape.IsMklTensor()) { in AllocateOutputTensor()
|
| D | mkl_fused_batch_norm_op.cc | 528 if (dnn_shape_src.IsMklTensor()) { in Compute()
573 if (dnn_shape_src.IsMklTensor()) in Compute()
599 if (dnn_shape_src.IsMklTensor()) { in Compute()
611 dnn_shape_src.IsMklTensor() in Compute()
615 auto src_md = dnn_shape_src.IsMklTensor() in Compute()
665 auto ndims = dnn_shape_src.IsMklTensor() ? dnn_shape_src.GetDimension() in Compute()
853 if (dnn_shape_diff_dst.IsMklTensor()) { in Compute()
867 if (dnn_shape_src.IsMklTensor()) { in Compute()
901 if (dnn_shape_src.IsMklTensor()) { in Compute()
903 } else if (dnn_shape_diff_dst.IsMklTensor()) { in Compute()
[all …]
|
| D | mkl_relu_op.cc | 473 if (dnn_shape_src.IsMklTensor()) { in Compute()
507 if (dnn_shape_src.IsMklTensor()) { in Compute()
589 if (!dnn_shape_src.IsMklTensor() && !dnn_shape_diff_dst.IsMklTensor()) { in Compute()
594 } else if (dnn_shape_src.IsMklTensor() && in Compute()
595 !dnn_shape_diff_dst.IsMklTensor()) { in Compute()
606 } else if (!dnn_shape_src.IsMklTensor() && in Compute()
607 dnn_shape_diff_dst.IsMklTensor()) { in Compute()
633 if (dnn_shape_src.IsMklTensor() || dnn_shape_diff_dst.IsMklTensor()) { in Compute()
634 common_md = dnn_shape_src.IsMklTensor() ? src_md : diff_dst_md; in Compute()
669 if (dnn_shape_src.IsMklTensor() || dnn_shape_diff_dst.IsMklTensor()) { in Compute()
[all …]
|
| D | mkl_conv_grad_bias_ops.cc | 68 bool input_is_mkl = mkl_context.input_shape.IsMklTensor(); in Compute()
168 bool input_is_mkl = input_shape.IsMklTensor(); in MklCreateInputLayouts()
245 bool input_is_mkl = input_shape.IsMklTensor(); in MklCleanup()
|
| D | mkl_concat_op.cc | 183 const TensorShape expected_shape = mkl_input_shapes[0].IsMklTensor() in Compute()
192 s.IsMklTensor() ? s.GetTfShape() : input_tensors[i].shape(); in Compute()
215 if (s.IsMklTensor()) in Compute()
414 dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() : dst_md; in Compute()
455 if (mkl_input_shapes[i].IsMklTensor()) { in CallEigenVersion()
|
| D | mkl_reshape_op.cc | 45 CHECK_EQ(mkl_shape_input.IsMklTensor(), true); in SkipReorder()
70 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); in Compute()
|
| D | mkl_identity_op.cc | 43 if (dnn_shape_input.IsMklTensor()) { in Compute()
|
| D | mkl_pooling_ops_common.h | 512 if (!original_input_mkl_shape.IsMklTensor()) { in InitMklPoolParameters()
614 input_mkl_shape.IsMklTensor() in ConfigureInput()
665 if (!input_mkl_shape.IsMklTensor()) { in SanityCheckInput()
721 input_gradient_mkl_shape.IsMklTensor() in ConfigureInputGradient()
743 input_gradient_mkl_shape.IsMklTensor() in ConfigureInputGradient()
|
| D | mkl_aggregate_ops.cc | 58 bool input1_in_mkl_format = src1_mkl_shape.IsMklTensor(); in Compute()
59 bool input2_in_mkl_format = src2_mkl_shape.IsMklTensor(); in Compute()
|
| D | mkl_conv_ops.cc | 444 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
449 CHECK(!mkl_filter_shape.IsMklTensor()) in Compute()
706 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateInputLayouts()
826 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup()
933 OP_REQUIRES(context, filter_mkl_shape.IsMklTensor() == false, in Compute()
1030 auto src_md = src_mkl_shape.IsMklTensor() in Compute()
1043 DCHECK(!filter_mkl_shape.IsMklTensor()); in Compute()
1045 filter_mkl_shape.IsMklTensor() in Compute()
1772 if (summand_mkl_shape.IsMklTensor()) { in AllocateOutputTensor()
1822 summand_mkl_shape.IsMklTensor() in AllocateOutputTensor()
|
| D | mkl_conv_grad_input_ops.cc | 385 filter_mkl_shape.IsMklTensor() in Compute()
396 diff_dst_mkl_shape.IsMklTensor() in Compute()
490 CHECK(!input_mkl_shape.IsMklTensor()) in ValidateMklShapes()
|
| D | mkl_tfconv_op.h | 71 if (!input_shape.IsMklTensor()) { in ConvertMklToTf()
|
| D | mkl_conv_grad_filter_ops.cc | 443 src_mkl_shape.IsMklTensor() in Compute()
451 diff_dst_mkl_shape.IsMklTensor() in Compute()
615 CHECK(!filter_mkl_shape.IsMklTensor()) in ValidateMklShapes()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 166 const bool IsMklTensor() const { return isMklTensor_; } in IsMklTensor() function
479 if (this->IsMklTensor() != input_shape.IsMklTensor()) {
485 if (this->IsMklTensor()) {
497 if (!this->IsMklTensor()) {
504 inline const bool IsMklTensor() const { return data_.is_mkl_tensor_; } in IsMklTensor() function
663 return IsMklTensor() ? GetMklLayout() : GetTfLayout(); in GetCurLayout()
795 if (!s.IsMklTensor()) { in AreAllMklTensors()
836 if (!mkl_shape.IsMklTensor()) in ConvertMklToTF()
951 if (input_mkl_shape.IsMklTensor()) { in GetTfShape()
|