Home
last modified time | relevance | path

Searched refs:MklDnnShape (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_relu_op.cc444 MklDnnShape dnn_shape_src; in Compute()
450 MklDnnShape dnn_shape_dst; in Compute()
598 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute()
609 MklDnnShape dnn_shape_diff_src; in Compute()
774 MklDnnShape dnn_shape_src; in Compute_Scalar()
780 MklDnnShape dnn_shape_dst; in Compute_Scalar()
809 MklDnnShape dnn_shape_diff_dst; in Compute_Scalar()
812 MklDnnShape dnn_shape_diff_src; in Compute_Scalar()
842 MklDnnShape dnn_shape_src; in Compute_Scalar()
848 MklDnnShape dnn_shape_dst; in Compute_Scalar()
[all …]
Dmkl_aggregate_ops.cc44 MklDnnShape src_mkl_shape; in GetTensorShape()
75 MklDnnShape src_mkl_shape; in FindMKLInputIndex()
91 MklDnnShape output_mkl_shape; in ComputeScalar()
116 MklDnnShape mkl_shape; in Compute()
137 MklDnnShape output_mkl_shape; in Compute()
184 MklDnnShape src_mkl_shape; in Compute()
Dmkl_input_conversion_op.cc71 MklDnnShape input_shape_0; in Compute()
75 MklDnnShape input_shape_1; in Compute()
123 MklDnnShape mkl_output_mkl_shape; in Compute()
192 const MklDnnShape* mkl_shape; in Compute()
224 MklDnnShape mkl_output_mkl_shape; in Compute()
Dmkl_lrn_op.cc98 MklDnnShape src_dnn_shape; in Compute()
197 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
231 MklDnnShape mkl_output_mkl_shape; in MklDefaultToEigen()
239 MklDnnShape workspace_mkl_shape; in MklDefaultToEigen()
269 MklDnnShape workspace_mkl_shape; in AllocateWorkspaceTensor()
283 MklDnnShape src_dnn_shape; in SanityCheckInputs()
345 MklDnnShape input_grad_dnn_shape, orig_input_dnn_shape, in Compute()
459 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
477 const MklDnnShape& input_grad_dnn_shape, in ConfigureInputGradient()
507 MklDnnShape input_grad_dnn_shape, orig_input_dnn_shape, in MklDefaultToEigen()
[all …]
Dmkl_quantize_op.cc269 MklDnnShape output_mkl_shape; in ComputeScalar()
276 MklDnnShape min_mkl_shape; in ComputeScalar()
282 MklDnnShape max_mkl_shape; in ComputeScalar()
332 MklDnnShape src_mkl_shape; in Compute()
405 MklDnnShape output_mkl_shape; in Compute()
426 MklDnnShape min_mkl_shape; in Compute()
432 MklDnnShape max_mkl_shape; in Compute()
Dmkl_reshape_op.cc46 bool SkipReorder(const MklDnnShape& mkl_shape_input, in SkipReorder()
65 MklDnnShape mkl_shape_input; in Compute()
160 MklDnnShape mkl_shape_output; in Compute()
Dmkl_conv_grad_input_ops.cc319 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute()
350 MklDnnShape diff_src_mkl_shape; in Compute()
443 MklDnnShape diff_src_mkl_shape; in Compute()
511 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
512 const MklDnnShape& filter_mkl_shape, in ValidateMklShapes()
513 const MklDnnShape& obp_mkl_shape) { in ValidateMklShapes()
570 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
Dmkl_slice_op.cc73 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs()
146 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs()
386 MklDnnShape output_mkl_shape; in ComputeMklSlice()
401 MklDnnShape input_mkl_shape; in ComputeMklSlice()
470 const MklDnnShape& input_mkl_shape, in AllocateOutputTensor()
474 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor()
Dmkl_matmul_op_fused.cc63 MklDnnShape src_mkl_shape; in Compute()
64 MklDnnShape weight_mkl_shape; in Compute()
137 MklDnnShape output_mkl_shape; in Compute()
144 MklDnnShape add_mkl_shape; in Compute()
Dmkl_conv_grad_filter_ops.cc376 MklDnnShape src_mkl_shape, filter_mkl_shape, diff_dst_mkl_shape; in Compute()
398 MklDnnShape diff_filter_mkl_shape; in Compute()
482 MklDnnShape diff_filter_mkl_shape; in Compute()
623 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes()
624 const MklDnnShape& filter_mkl_shape, in ValidateMklShapes()
625 const MklDnnShape& obp_mkl_shape) { in ValidateMklShapes()
685 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
702 MklDnnShape bias_grad_mkl_shape; in AllocateBiasGradTensor()
Dmkl_pooling_ops_common.h422 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
500 const MklDnnShape& original_input_mkl_shape, in InitMklPoolParameters()
548 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor()
607 const MklDnnShape& input_mkl_shape, in ConfigureInput()
651 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
670 const MklDnnShape& input_mkl_shape) { in SanityCheckInput()
703 MklDnnShape output_mkl_shape; in AllocateOutputTensor()
Dmkl_dequantize_op.cc63 MklDnnShape src_mkl_shape; in Compute()
93 MklDnnShape output_mkl_shape; in Compute()
Dmkl_maxpooling_op.cc58 MklDnnShape dnn_shape_input; in Compute()
170 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
210 MklDnnShape workspace_mkl_shape; in AllocateWorkspaceTensor()
243 MklDnnShape orig_input_mkl_shape, grad_mkl_shape; in Compute()
Dmkl_fused_batch_norm_op.cc692 MklDnnShape dnn_shape_src; in Compute()
846 MklDnnShape dnn_shape_dst; in Compute()
945 MklDnnShape dnn_shape_dst; in HandleEmptyInput()
983 MklDnnShape mkl_shape_batch_mean; in AllocateTFOutputs()
996 MklDnnShape mkl_shape_batch_variance; in AllocateTFOutputs()
1008 MklDnnShape mkl_shape_saved_mean; in AllocateTFOutputs()
1019 MklDnnShape mkl_shape_saved_variance; in AllocateTFOutputs()
1034 MklDnnShape mkl_shape_reserved_space; in AllocateTFOutputs()
1080 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute()
1244 MklDnnShape dnn_shape_diff_src; in Compute()
[all …]
Dmkl_identity_op.cc38 MklDnnShape dnn_shape_input; in Compute()
Dmkl_concat_op.cc461 std::vector<MklDnnShape> mkl_input_shapes(N); in Compute()
721 MklDnnShape dnn_shape_dst; in Compute()
760 MklDnnShape dnn_shape_dst; in Compute()
784 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
797 MklDnnShape dnn_shape_dst; in Compute()
847 MklDnnShape output_data_mkl_shape; in CallEigenVersion()
852 MklDnnShape output_min_max_mkl_shape; in CallEigenVersion()
Dmkl_avgpooling_op.cc54 MklDnnShape dnn_shape_input; in Compute()
145 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
184 MklDnnShape orig_input_mkl_shape, grad_mkl_shape; in Compute()
Dmkl_conv_ops.cc502 MklDnnShape src_mkl_shape, filter_mkl_shape; in Compute()
558 MklDnnShape dst_mkl_shape; in Compute()
664 MklDnnShape output_mkl_shape; in Compute()
852 MklDnnShape* output_mkl_shape, in AllocateOutputTensor()
878 MklDnnShape add_mkl_shape; in AllocateOutputTensor()
966 MklTensorFormat GetFilterTfDataFormat(const MklDnnShape* filter_mkl_shape, in GetFilterTfDataFormat()
977 const MklDnnShape* filter_mkl_shape) { in AllocatePersistentTensor()
1017 MklDnnShape filter_mkl_shape; in AllocateFilterOutputTensor()
1094 const MklDnnShape& filter_mkl_shape) TF_LOCKS_EXCLUDED(mu_) { in CacheFilter()
1389 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
[all …]
Dmkl_softmax_op.cc206 MklDnnShape src_mkl_shape; in Compute()
277 MklDnnShape output_mkl_shape; in Compute()
Dmkl_tfconv_op.h70 MklDnnShape input_shape; in ConvertMklToTf()
Dmkl_dequantize_op_test.cc66 MklDnnShape mkl_shape; in CreateMklInput()
Dmkl_qmatmul_op.cc167 MklDnnShape src_mkl_shape, weight_mkl_shape; in Compute()
324 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute()
Dmkl_pooling_ops_common.cc256 const MklDnnShape* mklInputShape) { in Init()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h247 class MklDnnShape {
269 MklDnnShape() { in MklDnnShape() function
279 ~MklDnnShape() {} in ~MklDnnShape()
280 TF_DISALLOW_COPY_AND_ASSIGN(MklDnnShape); // Cannot copy
284 inline bool operator==(const MklDnnShape& input_shape) const {
603 typedef std::vector<MklDnnShape> MklDnnShapeList;
625 const MklDnnShape& input_mkl_shape, in ConvertMklToTF()
677 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape, in GetMklShape()
693 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape) { in GetMklShape()
729 MklDnnShape input_mkl_shape;
[all …]
Dmkl_util_test.cc32 MklDnnShape a_mkldnn_shape; in TEST()
44 MklDnnShape b_mkldnn_shape; in TEST()

12