colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HLO Random operations should match shape constraints. 

PiperOrigin-RevId: 323844002
This commit is contained in:
Robert Suderman 2020-07-29 12:31:15 -07:00 committed by TensorFlow MLIR Team
parent 1b0eb4baa7
commit de5ddaf7c9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
View File

@ -1324,7 +1324,7 @@ def HLO_RngUniformOp : HLO_Op<"rng_uniform", []>, BASE_HLO_RngUniformOp {
let arguments = (ins
HLO_PredIntOrFpTensor:$a,
HLO_PredIntOrFpTensor:$b,
I64Tensor:$shape
HLO_DimensionTensor:$shape
);
let results = (outs HLO_PredIntOrFpTensor);
@ -1336,7 +1336,7 @@ def HLO_RngNormalOp : HLO_Op<"rng_normal", []>, BASE_HLO_RngNormalOp {
let arguments = (ins
HLO_FpTensor:$mu,
HLO_FpTensor:$sigma,
I64Tensor:$shape
HLO_DimensionTensor:$shape
);
let results = (outs HLO_FpTensor);