| commit | a2c5274a5f9dae02fb669c7529368135e5676dd9 | [log] [tgz] |
|---|---|---|
| author | Alexander Belyaev <pifon@google.com> | Mon Dec 13 13:02:58 2021 -0800 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Mon Dec 13 13:08:34 2021 -0800 |
| tree | 2bdd25291c4bcd6f9f3490886d1400cd26abd36e | |
| parent | a6184576ffcc3492340646aedde31b32f32b4bbd [diff] |
[hlo] Update comments that mention `linalg.tensor_expand/collapse_shape`. PiperOrigin-RevId: 416114699 Change-Id: If02d3dab11ae5762a2df1a9f7fa20ebcdaa791e3
diff --git a/tensorflow/compiler/mlir/hlo/lib/Transforms/reshape_simplifier.cc b/tensorflow/compiler/mlir/hlo/lib/Transforms/reshape_simplifier.cc index 2dda6b4..e794be6 100644 --- a/tensorflow/compiler/mlir/hlo/lib/Transforms/reshape_simplifier.cc +++ b/tensorflow/compiler/mlir/hlo/lib/Transforms/reshape_simplifier.cc
@@ -56,7 +56,7 @@ } // Rewrite dynamic reshapes that only insert one dimensions into -// linalg.tensor_expand_shape. +// tensor.expand_shape. struct ReshapeToExpandShape final : public OpRewritePattern<mhlo::DynamicReshapeOp> { ReshapeToExpandShape(MLIRContext *ctx) : OpRewritePattern(ctx) {}
diff --git a/tensorflow/compiler/mlir/tfrt/jit/transforms/tf_cpurt_codegen_reduction.cc b/tensorflow/compiler/mlir/tfrt/jit/transforms/tf_cpurt_codegen_reduction.cc index 31a1807..dc470fb 100644 --- a/tensorflow/compiler/mlir/tfrt/jit/transforms/tf_cpurt_codegen_reduction.cc +++ b/tensorflow/compiler/mlir/tfrt/jit/transforms/tf_cpurt_codegen_reduction.cc
@@ -129,7 +129,7 @@ // : tensor<?xf32> to tensor<?xf32> // %tile_pad = linalg.pad_tensor %tile // : tensor<?xf32> to tensor<VECTOR_SIZExf32> -// %tile_reshape = linalg.tensor_expand_shape %tile_pad [[0, 1]] +// %tile_reshape = tensor.expand_shape %tile_pad [[0, 1]] // : tensor<VECTOR_SIZExf32> into tensor<1xVECTOR_SIZExf32> // %combine = linalg.generic ins(%tile_reshape : tensor<1xVECTOR_SIZExf32>) // outs(%tmp_result_ : tensor<VECTOR_SIZExf32>) -> tensor<VECTOR_SIZExf32>