def _as_graph_element(self):
if self.quantizer and context.should_quantize():
return self.quantizer(self.latent_variable)
graph_element = self.latent_variable._as_graph_element()
if graph_element is None:
return self._op
return graph_element
def _quantize(self, value):
if self.quantizer and context.should_quantize():
return self.quantizer(value)
return value
def test_quantize_scope(quantized):
assert context.should_quantize() == quantized