Searched refs:GetInitTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 88 Value GetInitTensor(OpBuilder& b, Location loc, ShapedType type, in GetInitTensor() function
194 GetInitTensor(rewriter, loc, result_type, dyn_sizes)); in matchAndRewrite()
393 : ValueRange{GetInitTensor(rewriter, loc, result_type, dyn_sizes)}, in matchAndRewrite()
862 : ValueRange{GetInitTensor(rewriter, loc, result_shaped_type, in matchAndRewrite()
1146 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite()
1234 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite()
1353 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite()
|