Sha256: d7a749ae5d58401f02fda13dcc6adcdb290c72c65c2d5335ed630e4047d33f25
Contents?: true
Size: 502 Bytes
Versions: 9
Compression:
Stored size: 502 Bytes
Contents
TensorStream::OpMaker.define_operation :min do |op|
op.what_it_does "Returns the min of x and y (i.e. x < y ? x : y) element-wise."
op.parameter :input_a, "tensor X", nil, validate: 'NUMERIC_TYPES'
op.parameter :input_b, "tensor Y", nil, validate: 'NUMERIC_TYPES'
op.apply_data_type_coercion!
op.supports_broadcasting!
op.option :name, "Optional name", :nil
op.define_gradient do |grad, node, params|
_min_or_max_grad(node.inputs, grad, ->(a, b) { ts.less_equal(a, b) })
end
end
Version data entries
9 entries across 9 versions & 1 rubygems