Sha256: 277e52bb1209488b6888f9f447fef97342b5399ffc4ebcc409bee166d5baf48e
Contents?: true
Size: 436 Bytes
Versions: 5
Compression:
Stored size: 436 Bytes
Contents
TensorStream::OpMaker.define_operation :less_equal do |op|
op.what_it_does "Returns the truth value of (x <= y) element-wise."
op.parameter :input_a, "tensor X"
op.parameter :input_b, "tensor Y"
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.greater_equal(a, b) })
end
end
Version data entries
5 entries across 5 versions & 1 rubygems