Sha256: 34d39fe73d1ac7de6fca71db3232d1df01daf6fcdf3d3831225f9ee073d2dc92
Contents?: true
Size: 481 Bytes
Versions: 4
Compression:
Stored size: 481 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 op.define_data_type do :boolean end end
Version data entries
4 entries across 4 versions & 1 rubygems