Searched refs:GetTfDataFormat (Results 1 – 12 of 12) sorted by relevance
108 if (input_shape_0.GetTfDataFormat() == in Compute()109 input_shape_1.GetTfDataFormat()) { in Compute()128 input_shape_0.GetTfDataFormat()); in Compute()229 mkl_shape->GetTfDataFormat()); in Compute()
172 mkl_data_format = mkl_shape.GetTfDataFormat(); in Compute()227 mkl_shape.GetTfDataFormat()); in Compute()
624 mkl_input_shapes[k].GetTfDataFormat()); in Compute()670 auto orig_tf_format = mkl_input_shapes[0].GetTfDataFormat(); in Compute()728 mkl_input_shapes[0].GetTfDataFormat()); in Compute()883 MklTensorFormatToMklDnnDataFormat(input_shapes[k].GetTfDataFormat())); in FindMklCommonFormat()891 MklTensorFormatToMklDnnDataFormat(input_shapes[0].GetTfDataFormat())); in FindMklCommonFormat()
114 src_mkl_shape.GetTfDataFormat()); in Compute()
506 dnn_shape_src.GetTfDataFormat()); in Compute()635 MklTensorFormat src_mkl_data_format = dnn_shape_src.GetTfDataFormat(); in Compute()648 dnn_shape_diff_dst.GetTfDataFormat(); in Compute()717 dnn_shape_src.GetTfDataFormat()); in Compute()722 dnn_shape_diff_dst.GetTfDataFormat()); in Compute()
56 return (mkl_shape_input.GetTfDataFormat() == in SkipReorder()
405 auto input_mkl_format = input_mkl_shape.GetTfDataFormat(); in ComputeMklSlice()489 input_mkl_shape.GetTfDataFormat()); in AllocateOutputTensor()
155 auto input_format = src_dnn_shape.GetTfDataFormat(); in Compute()411 auto orig_input_format = orig_input_dnn_shape.GetTfDataFormat(); in Compute()
413 src_mkl_shape.GetTfDataFormat()); in Compute()
898 output_mkl_shape->GetTfDataFormat()); in AllocateOutputTensor()969 return filter_mkl_shape->GetTfDataFormat(); in GetFilterTfDataFormat()
53 #define GET_TF_DATA_FORMAT(shape, mem_desc) shape.GetTfDataFormat()116 input_mkl_shape.GetTfDataFormat() == MKL_TENSOR_FORMAT_BLOCKED
434 inline MklTensorFormat GetTfDataFormat() const { in GetTfDataFormat() function