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

Version Path
tensor_stream-1.0.9 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.8 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.7 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.6 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.5 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.4 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.3 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.2 lib/tensor_stream/ops/floor_div.rb
tensor_stream-1.0.1 lib/tensor_stream/ops/floor_div.rb