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

Version Path
tensor_stream-1.0.9 lib/tensor_stream/ops/less_equal.rb
tensor_stream-1.0.8 lib/tensor_stream/ops/less_equal.rb
tensor_stream-1.0.7 lib/tensor_stream/ops/less_equal.rb
tensor_stream-1.0.6 lib/tensor_stream/ops/less_equal.rb