Sha256: ae73b451e70798d4160dc53390383dc1fd78ba4b8b110354b5cb705b23b76bce
Contents?: true
Size: 505 Bytes
Versions: 9
Compression:
Stored size: 505 Bytes
Contents
TensorStream::OpMaker.define_operation :max do |op| op.what_it_does "Returns the max 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.greater_equal(a, b) }) end end
Version data entries
9 entries across 9 versions & 1 rubygems