提交 a486ae68 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

In Scan.infer_shape, the shapes of node.inputs are valid values to use.

上级 b3965fab
......@@ -628,9 +628,10 @@ class Scan(Op):
inputs = self.inputs,
input_shapes = inner_ins_shapes)
# Will be used to check if outs_shape can be expressed without using
# variables in self.inputs
# variables in self.inputs.
# The shapes of node.inputs are valid.
validator = scan_utils.Validator(
valid = [],
valid = input_shapes,
invalid = self.inputs,
valid_equivalent = out_equivalent)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论