Home
last modified time | relevance | path

Searched refs:data_format_tf (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.h79 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf,
86 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf,
91 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf,
Dlegalize_utils.cc236 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf, in getPaddingValuesFromPadType() argument
249 4, data_format_tf, i); // 4D tensor, NHWC/NCHW format in getPaddingValuesFromPadType()
284 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf, in getPaddingValuesFromExplicitPadAttr() argument
290 int64_t dim = GetTensorSpatialDimIndex(4, data_format_tf, in getPaddingValuesFromExplicitPadAttr()
304 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf, in getTransposeConv2dPaddingValues() argument
319 4, data_format_tf, i); // 4D tensor, NHWC/NCHW format in getTransposeConv2dPaddingValues()
321 4, data_format_tf, i); // 4D tensor, NHWC/NCHW format in getTransposeConv2dPaddingValues()
Dlegalize_tf.cc896 tensorflow::TensorFormat data_format_tf; in matchAndRewrite() local
897 if (!FormatFromString(tf_dwconv2d_op.data_format().str(), &data_format_tf)) in matchAndRewrite()
902 tf_dwconv2d_op.explicit_paddings(), data_format_tf, rewriter); in matchAndRewrite()
904 if (!getPaddingValuesFromPadType(tf_pad, data_format_tf, in matchAndRewrite()
987 tensorflow::TensorFormat data_format_tf; in matchAndRewrite() local
988 if (!FormatFromString(tf_conv_op.data_format().str(), &data_format_tf)) in matchAndRewrite()
993 tf_conv_op.explicit_paddings(), data_format_tf, rewriter); in matchAndRewrite()
995 if (!getTransposeConv2dPaddingValues(tf_pad, data_format_tf, in matchAndRewrite()
Dlegalize_common.cc2842 tensorflow::TensorFormat data_format_tf; in convertTFConv2DCommon() local
2843 if (!FormatFromString(data_format_ref.str(), &data_format_tf)) in convertTFConv2DCommon()
2848 data_format_tf, rewriter); in convertTFConv2DCommon()
2850 if (!getPaddingValuesFromPadType(tf_pad, data_format_tf, in convertTFConv2DCommon()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md139 tensorflow::TensorFormat data_format_tf)
145 int64_t dim = GetTensorSpatialDimIndex(4, data_format_tf, i)