Searched refs:allow_scalar_outputs (Results 1 – 7 of 7) sorted by relevance
382 not fake_mode.shape_env.allow_scalar_outputs383 and not fake_mode.allow_scalar_outputs
1133 self.allow_scalar_outputs = False
2301 allow_scalar_outputs: bool2380 allow_scalar_outputs=True, argument2423 allow_scalar_outputs=allow_scalar_outputs,2593 def allow_scalar_outputs(self): member in ShapeEnv2594 return self.settings.allow_scalar_outputs
442 allow_scalar_outputs=False, allow_dynamic_output_shape_ops=False
298 allow_scalar_outputs=config.capture_scalar_outputs,
3486 allow_scalar_outputs=config.capture_scalar_outputs,3730 allow_scalar_outputs=config.capture_scalar_outputs,
9309 main, other = ShapeEnv(allow_scalar_outputs=False), ShapeEnv()