Home
last modified time | relevance | path

Searched refs:GetTfDataFormat (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_aggregate_ops.cc138 memory::format src1_mkl_data_format = src1_mkl_shape.GetTfDataFormat(); in Compute()
152 memory::format src2_mkl_data_format = src2_mkl_shape.GetTfDataFormat(); in Compute()
225 src1_mkl_shape.GetTfDataFormat()); in Compute()
229 src2_mkl_shape.GetTfDataFormat()); in Compute()
Dmkl_reshape_op.cc55 if (mkl_shape_input.GetTfDataFormat() == input_mkl_md.data.format && in SkipReorder()
56 mkl_shape_input.GetTfDataFormat() != memory::format::blocked) { in SkipReorder()
Dmkl_input_conversion_op.cc129 input_shape_0.GetTfDataFormat()); in Compute()
227 mkl_shape->GetTfDataFormat()); in Compute()
Dmkl_relu_op.cc514 dnn_shape_src.GetTfDataFormat()); in Compute()
599 memory::format src_mkl_data_format = dnn_shape_src.GetTfDataFormat(); in Compute()
611 dnn_shape_diff_dst.GetTfDataFormat(); in Compute()
677 dnn_shape_src.GetTfDataFormat()); in Compute()
682 dnn_shape_diff_dst.GetTfDataFormat()); in Compute()
Dmkl_slice_op.cc396 auto input_mkl_format = input_mkl_shape.GetTfDataFormat(); in ComputeMklSlice()
463 input_mkl_shape.GetTfDataFormat()); in AllocateOutputTensor()
Dmkl_concat_op.cc354 auto orig_tf_format = mkl_input_shapes[0].GetTfDataFormat(); in Compute()
403 mkl_input_shapes[0].GetTfDataFormat()); in Compute()
Dmkl_lrn_op.cc150 memory::format input_format = src_dnn_shape.GetTfDataFormat(); in Compute()
413 memory::format orig_input_format = orig_input_dnn_shape.GetTfDataFormat(); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h628 inline memory::format GetTfDataFormat() const { in GetTfDataFormat() function