Sha256: e3a325542ab193d081c7c887319289a0f286bb3752b4eb42cd074ad2cd2091b0
Contents?: true
Size: 363 Bytes
Versions: 9
Compression:
Stored size: 363 Bytes
Contents
TensorStream::OpMaker.define_operation :floor_div do |op| op.what_it_does "Returns element-wise integer divistion." 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| [nil, nil] end end
Version data entries
9 entries across 9 versions & 1 rubygems