| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_relu_op.cc | 461 MklDnnShape dnn_shape_src; in Compute()
505 MklDnnShape dnn_shape_dst; in Compute()
575 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute()
667 MklDnnShape dnn_shape_diff_src; in Compute()
731 MklDnnShape dnn_shape_src; in Compute_Scalar()
737 MklDnnShape dnn_shape_dst; in Compute_Scalar()
764 MklDnnShape dnn_shape_diff_dst; in Compute_Scalar()
767 MklDnnShape dnn_shape_diff_src; in Compute_Scalar()
794 MklDnnShape dnn_shape_src; in Compute_Scalar()
800 MklDnnShape dnn_shape_dst; in Compute_Scalar()
[all …]
|
| D | mkl_input_conversion_op.cc | 73 MklDnnShape input_shape_0; in Compute()
77 MklDnnShape input_shape_1; in Compute()
124 MklDnnShape mkl_output_mkl_shape; in Compute()
190 const MklDnnShape* mkl_shape; in Compute()
222 MklDnnShape mkl_output_mkl_shape; in Compute()
|
| D | mkl_lrn_op.cc | 96 MklDnnShape src_dnn_shape; in Compute()
202 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
237 MklDnnShape mkl_output_mkl_shape; in MklDefaultToEigen()
245 MklDnnShape workspace_mkl_shape; in MklDefaultToEigen()
276 MklDnnShape workspace_mkl_shape; in AllocateWorkspaceTensor()
290 MklDnnShape src_dnn_shape; in SanityCheckInputs()
349 MklDnnShape input_grad_dnn_shape, orig_input_dnn_shape, in Compute()
452 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
470 const MklDnnShape& input_grad_dnn_shape, in ConfigureInputGradient()
527 MklDnnShape input_grad_dnn_shape, orig_input_dnn_shape, in MklDefaultToEigen()
[all …]
|
| D | mkl_reshape_op.cc | 43 bool SkipReorder(const MklDnnShape& mkl_shape_input, in SkipReorder()
68 MklDnnShape mkl_shape_input; in Compute()
155 MklDnnShape mkl_shape_output; in Compute()
186 MklDnnShape mkl_shape_output; in Compute()
|
| D | mkl_aggregate_ops.cc | 55 MklDnnShape src1_mkl_shape, src2_mkl_shape; in Compute()
82 MklDnnShape mkl_shape_dst; in Compute()
96 MklDnnShape mkl_shape_dst; in Compute()
214 MklDnnShape output_mkl_shape; in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 321 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute()
342 MklDnnShape diff_src_mkl_shape; in Compute()
424 MklDnnShape diff_src_mkl_shape; in Compute()
484 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
485 const MklDnnShape& filter_mkl_shape, in ValidateMklShapes()
486 const MklDnnShape& obp_mkl_shape) { in ValidateMklShapes()
544 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
|
| D | mkl_conv_grad_filter_ops.cc | 384 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute()
405 MklDnnShape diff_filter_mkl_shape; in Compute()
484 MklDnnShape diff_filter_mkl_shape; in Compute()
612 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
613 const MklDnnShape& filter_mkl_shape, in ValidateMklShapes()
614 const MklDnnShape& obp_mkl_shape) { in ValidateMklShapes()
677 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
695 MklDnnShape bias_grad_mkl_shape; in AllocateBiasGradTensor()
|
| D | mkl_slice_op.cc | 76 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs()
148 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs()
378 MklDnnShape output_mkl_shape; in ComputeMklSlice()
392 MklDnnShape input_mkl_shape; in ComputeMklSlice()
444 const MklDnnShape& input_mkl_shape, in AllocateOutputTensor()
448 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor()
|
| D | mkl_fused_batch_norm_op.cc | 525 MklDnnShape dnn_shape_src; in Compute()
659 MklDnnShape dnn_shape_dst; in Compute()
741 MklDnnShape dnn_shape_dst; in HandleEmptyInput()
774 MklDnnShape mkl_shape_batch_mean; in AllocateTFOutputs()
785 MklDnnShape mkl_shape_batch_variance; in AllocateTFOutputs()
797 MklDnnShape mkl_shape_saved_mean; in AllocateTFOutputs()
806 MklDnnShape mkl_shape_saved_variance; in AllocateTFOutputs()
848 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute()
992 MklDnnShape dnn_shape_diff_src; in Compute()
1057 MklDnnShape dnn_shape_diff_src; in HandleEmptyInput()
[all …]
|
| D | mkl_pooling_ops_common.h | 434 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
510 const MklDnnShape& original_input_mkl_shape, in InitMklPoolParameters()
558 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor()
605 const MklDnnShape& input_mkl_shape, in ConfigureInput()
648 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
664 const MklDnnShape& input_mkl_shape) { in SanityCheckInput()
700 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
715 const MklDnnShape& input_gradient_mkl_shape, in ConfigureInputGradient()
|
| D | mkl_maxpooling_op.cc | 516 MklDnnShape dnn_shape_input; in Compute()
621 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
663 MklDnnShape workspace_mkl_shape; in AllocateWorkspaceTensor()
695 MklDnnShape orig_input_mkl_shape, grad_mkl_shape; in Compute()
817 const MklDnnShape& orig_input_mkl_shape, in SanityCheckInputs()
818 const MklDnnShape& orig_output_mkl_shape, in SanityCheckInputs()
819 const MklDnnShape& grad_mkl_shape, in SanityCheckInputs()
820 const MklDnnShape& workspace_mkl_shape) { in SanityCheckInputs()
|
| D | mkl_softmax_op.cc | 54 MklDnnShape src_mkl_shape; in Compute()
137 MklDnnShape output_mkl_shape; in Compute()
|
| D | mkl_avgpooling_op.cc | 447 MklDnnShape dnn_shape_input; in Compute()
544 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
581 MklDnnShape orig_input_mkl_shape, grad_mkl_shape; in Compute()
675 const MklDnnShape& original_input_mkl_shape, in SanityCheckInputs()
676 const MklDnnShape& input_gradient_mkl_shape) { in SanityCheckInputs()
|
| D | mkl_concat_op.cc | 163 std::vector<MklDnnShape> mkl_input_shapes(N); in Compute()
394 MklDnnShape dnn_shape_dst; in Compute()
426 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
472 MklDnnShape dnn_shape_output; in CallEigenVersion()
|
| D | mkl_identity_op.cc | 39 MklDnnShape dnn_shape_input; in Compute()
|
| D | mkl_tfconv_op.h | 67 MklDnnShape input_shape; in ConvertMklToTf()
|
| D | mkl_conv_ops.cc | 930 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute()
977 MklDnnShape dst_mkl_shape; in Compute()
1264 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
1336 MklDnnShape filter_mkl_shape; in AllocateFilterOutputTensor()
1566 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
1769 MklDnnShape summand_mkl_shape; in AllocateOutputTensor()
1800 MklDnnShape summand_mkl_shape; in AllocateOutputTensor()
|
| D | mkl_pooling_ops_common.cc | 271 const MklDnnShape* mklInputShape) { in Init()
|
| D | mkl_conv_ops_test.cc | 108 MklDnnShape non_mkl_shape; in NonMklTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 33 MklDnnShape a_mkldnn_shape; in TEST()
45 MklDnnShape b_mkldnn_shape; in TEST()
|
| D | mkl_util.h | 423 class MklDnnShape {
445 MklDnnShape() { in MklDnnShape() function
455 ~MklDnnShape() {} in ~MklDnnShape()
456 TF_DISALLOW_COPY_AND_ASSIGN(MklDnnShape); // Cannot copy
478 inline bool operator==(const MklDnnShape& input_shape) const {
786 typedef std::vector<MklDnnShape> MklDnnShapeList;
833 const MklDnnShape& mkl_shape) { in ConvertMklToTF()
887 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape) { in GetMklShape()
949 MklDnnShape input_mkl_shape; in GetTfShape()
980 const MklDnnShape& mkl_shape) { in AllocateOutputSetMklShape()
[all …]
|