Home
last modified time | relevance | path

Searched refs:tf_pad (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,
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
240 assert(tf_pad != tensorflow::Padding::EXPLICIT); in getPaddingValuesFromPadType()
259 dim_dilation, dim_stride, tf_pad, &op_size, &pad_before_tf, in getPaddingValuesFromPadType()
304 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf, in getTransposeConv2dPaddingValues() argument
309 assert(tf_pad != tensorflow::Padding::EXPLICIT); in getTransposeConv2dPaddingValues()
Dlegalize_tf.cc533 tensorflow::Padding tf_pad; in matchAndRewrite() local
534 if (!GetPaddingFromString(tf_avgpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
551 tf_pad, in matchAndRewrite()
603 tensorflow::Padding tf_pad; in matchAndRewrite() local
604 if (!GetPaddingFromString(tf_maxpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
621 tf_pad, in matchAndRewrite()
892 tensorflow::Padding tf_pad; in matchAndRewrite() local
893 if (!GetPaddingFromString(tf_dwconv2d_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
900 if (tf_pad == tensorflow::Padding::EXPLICIT) { in matchAndRewrite()
904 if (!getPaddingValuesFromPadType(tf_pad, data_format_tf, in matchAndRewrite()
[all …]
Dlegalize_tfl.cc1020 tensorflow::Padding tf_pad; in matchAndRewrite() local
1021 if (!GetPaddingFromString(tfl_avgpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
1032 tf_pad, in matchAndRewrite()
1074 tensorflow::Padding tf_pad; in matchAndRewrite() local
1075 if (!GetPaddingFromString(tfl_maxpool_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
1086 tf_pad, in matchAndRewrite()
1141 tensorflow::Padding tf_pad; in matchAndRewrite() local
1142 if (!GetPaddingFromString(tfl_conv2d_op.padding().str(), &tf_pad).ok()) in matchAndRewrite()
1147 tf_pad, in matchAndRewrite()
1230 tensorflow::Padding tf_pad; in matchAndRewrite() local
[all …]
Dlegalize_common.cc2836 tensorflow::Padding tf_pad; in convertTFConv2DCommon() local
2837 if (!GetPaddingFromString(padding_ref.str(), &tf_pad).ok()) { in convertTFConv2DCommon()
2846 if (tf_pad == tensorflow::Padding::EXPLICIT) { in convertTFConv2DCommon()
2850 if (!getPaddingValuesFromPadType(tf_pad, data_format_tf, in convertTFConv2DCommon()