Home
last modified time | relevance | path

Searched refs:SetMklTensor (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_relu_op.cc508 dnn_shape_dst.SetMklTensor(true); in Compute()
517 dnn_shape_dst.SetMklTensor(false); in Compute()
671 dnn_shape_diff_src.SetMklTensor(true); in Compute()
686 dnn_shape_diff_src.SetMklTensor(false); in Compute()
738 dnn_shape_dst.SetMklTensor(false); in Compute_Scalar()
768 dnn_shape_diff_src.SetMklTensor(false); in Compute_Scalar()
801 dnn_shape_dst.SetMklTensor(false); in Compute_Scalar()
835 dnn_shape_diff_src.SetMklTensor(false); in Compute_Scalar()
873 dnn_shape_dst.SetMklTensor(false); in Compute_Scalar()
906 dnn_shape_diff_src.SetMklTensor(false); in Compute_Scalar()
[all …]
Dmkl_aggregate_ops.cc83 mkl_shape_dst.SetMklTensor(false); in Compute()
97 mkl_shape_dst.SetMklTensor(false); in Compute()
218 output_mkl_shape.SetMklTensor(true); in Compute()
233 output_mkl_shape.SetMklTensor(false); in Compute()
Dmkl_fused_batch_norm_op.cc661 dnn_shape_dst.SetMklTensor(true); in Compute()
742 dnn_shape_dst.SetMklTensor(false); in HandleEmptyInput()
775 mkl_shape_batch_mean.SetMklTensor(false); in AllocateTFOutputs()
786 mkl_shape_batch_variance.SetMklTensor(false); in AllocateTFOutputs()
798 mkl_shape_saved_mean.SetMklTensor(false); in AllocateTFOutputs()
807 mkl_shape_saved_variance.SetMklTensor(false); in AllocateTFOutputs()
994 dnn_shape_diff_src.SetMklTensor(true); in Compute()
1058 dnn_shape_diff_src.SetMklTensor(false); in HandleEmptyInput()
1084 mkl_shape_diff_scale.SetMklTensor(false); in AllocateTFOutputs()
1093 mkl_shape_diff_shift.SetMklTensor(false); in AllocateTFOutputs()
[all …]
Dmkl_softmax_op.cc145 output_mkl_shape.SetMklTensor(true); in Compute()
152 output_mkl_shape.SetMklTensor(false); in Compute()
Dmkl_concat_op.cc398 dnn_shape_dst.SetMklTensor(true); in Compute()
406 dnn_shape_dst.SetMklTensor(false); in Compute()
427 output_min_mkl_shape.SetMklTensor(false); in Compute()
428 output_max_mkl_shape.SetMklTensor(false); in Compute()
473 dnn_shape_output.SetMklTensor(false); in CallEigenVersion()
Dmkl_lrn_op.cc205 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
238 mkl_output_mkl_shape.SetMklTensor(false); in MklDefaultToEigen()
246 workspace_mkl_shape.SetMklTensor(false); in MklDefaultToEigen()
279 workspace_mkl_shape.SetMklTensor(false); in AllocateWorkspaceTensor()
455 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
568 mkl_output_mkl_shape.SetMklTensor(false); in MklDefaultToEigen()
Dmkl_reshape_op.cc156 mkl_shape_output.SetMklTensor(false); in Compute()
187 mkl_shape_output.SetMklTensor(true); in Compute()
Dmkl_maxpooling_op.cc107 mkl_out_shape.SetMklTensor(true); in Compute()
125 mkl_workspace_shape.SetMklTensor(false); in Compute()
279 mkl_output_shape.SetMklTensor(true); in Compute()
622 output_min_mkl_shape.SetMklTensor(false); in Compute()
623 output_max_mkl_shape.SetMklTensor(false); in Compute()
664 workspace_mkl_shape.SetMklTensor(false); in AllocateWorkspaceTensor()
Dmkl_input_conversion_op.cc125 mkl_output_mkl_shape.SetMklTensor(true); in Compute()
223 mkl_output_mkl_shape.SetMklTensor(true); in Compute()
Dmkl_avgpooling_op.cc126 mkl_out_shape.SetMklTensor(true); in Compute()
316 mkl_out_shape.SetMklTensor(true); in Compute()
545 output_min_mkl_shape.SetMklTensor(false); in Compute()
546 output_max_mkl_shape.SetMklTensor(false); in Compute()
Dmkl_conv_grad_filter_ops.cc406 diff_filter_mkl_shape.SetMklTensor(false); in Compute()
485 diff_filter_mkl_shape.SetMklTensor(false); in Compute()
678 output_mkl_shape.SetMklTensor(false); in AllocateOutputTensor()
696 bias_grad_mkl_shape.SetMklTensor(false); in AllocateBiasGradTensor()
Dmkl_conv_ops.cc543 mkl_output_mkl_shape.SetMklTensor(false); in Compute()
552 mkl_output_mkl_shape.SetMklTensor(false); in Compute()
627 mkl_output_mkl_shape.SetMklTensor(true); in Compute()
644 mkl_filter_output_mkl_shape.SetMklTensor(true); in Compute()
978 dst_mkl_shape.SetMklTensor(false); in Compute()
983 filter_mkl_shape.SetMklTensor(false); in Compute()
988 filter_mkl_shape.SetMklTensor(false); in Compute()
1265 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
1337 filter_mkl_shape.SetMklTensor(true); in AllocateFilterOutputTensor()
1567 output_min_mkl_shape.SetMklTensor(false); in Compute()
[all …]
Dmkl_conv_grad_input_ops.cc343 diff_src_mkl_shape.SetMklTensor(false); in Compute()
425 diff_src_mkl_shape.SetMklTensor(true); in Compute()
545 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
Dmkl_slice_op.cc459 output_mkl_shape->SetMklTensor(true); in AllocateOutputTensor()
468 output_mkl_shape->SetMklTensor(false); in AllocateOutputTensor()
Dmkl_pooling_ops_common.h559 output_mkl_shape.SetMklTensor(false); in AllocateEmptyOutputTensor()
649 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
701 output_mkl_shape.SetMklTensor(true); in AllocateOutputTensor()
Dmkl_conv_grad_bias_ops.cc93 output_mkl_shape.SetMklTensor(false); in Compute()
Dmkl_conv_ops_test.cc109 non_mkl_shape.SetMklTensor(false); in NonMklTensor()
/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc34 a_mkldnn_shape.SetMklTensor(true); in TEST()
46 b_mkldnn_shape.SetMklTensor(true); in TEST()
Dmkl_util.h168 void SetMklTensor(const bool isMklTensor) { isMklTensor_ = isMklTensor; } in SetMklTensor() function
505 inline void SetMklTensor(bool is_mkl_tensor) { in SetMklTensor() function
1172 mkl_shape_output.SetMklTensor(false); in CopyTfTensorInToOutWithShape()
1190 mkl_shape_output.SetMklTensor(false); in CopyTfTensorInToOutWithShape()
1209 mkl_shape_output.SetMklTensor(false); in ForwardTfTensorInToOut()
1228 dnn_shape_output.SetMklTensor(false); in ForwardTfTensorInToOut()
1262 mkl_shape_output.SetMklTensor(false); in SetDummyMklDnnShapeOutput()
1306 mkl_shape_output.SetMklTensor(false); in SetDummyMklShapeOutput()