Method: TensorStream::MathGradients._int32

Defined in:
lib/tensor_stream/math_gradients.rb

._int32(node, x) ⇒ Object



184
185
186
# File 'lib/tensor_stream/math_gradients.rb', line 184

def self._int32(node, x)
  (node.inputs[0].data_type == :int32 ? x : TensorStream.cast(x, :int32))
end