Searched refs:data_format_tf (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.h | 97 tensorflow::TensorFormat data_format_tf,
106 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf,
111 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf,
|
| D | legalize_utils.cc | 355 tensorflow::TensorFormat data_format_tf, in getPaddingValuesFromPadType() argument
371 4, data_format_tf, i); // 4D tensor, NHWC/NCHW format in getPaddingValuesFromPadType()
409 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf, in getPaddingValuesFromExplicitPadAttr() argument
415 int64_t dim = GetTensorSpatialDimIndex(4, data_format_tf, in getPaddingValuesFromExplicitPadAttr()
429 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf, in getTransposeConv2dPaddingValues() argument
445 4, data_format_tf, i); // 4D tensor, NHWC/NCHW format in getTransposeConv2dPaddingValues()
447 4, data_format_tf, i); // 4D tensor, NHWC/NCHW format in getTransposeConv2dPaddingValues()
|
| D | legalize_tf.cc | 899 tensorflow::TensorFormat data_format_tf; in matchAndRewrite() local
900 if (!FormatFromString(tf_dwconv2d_op.data_format().str(), &data_format_tf)) in matchAndRewrite()
905 tf_dwconv2d_op.explicit_paddings(), data_format_tf, rewriter); in matchAndRewrite()
907 if (!getPaddingValuesFromPadType(tf_pad, data_format_tf, in matchAndRewrite()
991 tensorflow::TensorFormat data_format_tf; in matchAndRewrite() local
992 if (!FormatFromString(tf_conv_op.data_format().str(), &data_format_tf)) in matchAndRewrite()
997 tf_conv_op.explicit_paddings(), data_format_tf, rewriter); in matchAndRewrite()
999 if (!getTransposeConv2dPaddingValues(tf_pad, data_format_tf, in matchAndRewrite()
|
| D | legalize_common.cc | 3372 tensorflow::TensorFormat data_format_tf; in convertTFConv2DCommon() local
3373 if (!FormatFromString(data_format_ref.str(), &data_format_tf)) in convertTFConv2DCommon()
3378 data_format_tf, rewriter); in convertTFConv2DCommon()
3380 if (!getPaddingValuesFromPadType(tf_pad, data_format_tf, in convertTFConv2DCommon()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 102 tensorflow::TensorFormat data_format_tf)
108 int64 dim = GetTensorSpatialDimIndex(4, data_format_tf, i)
|