Lines Matching defs:out
192 def elemwise_exp_add_on_buffers(lhs, rhs, out):
193 linalg.elemwise_unary(lhs, outs=[out])
194 linalg.elemwise_binary(out, rhs, outs=[out])
201 def elemwise_log_mul_on_buffers(lhs, rhs, out):
202 linalg.elemwise_unary(lhs, outs=[out], fun=UnaryFn.log)
203 linalg.elemwise_binary(out, rhs, outs=[out], fun=BinaryFn.mul)
235 def elemwise_exp_add_on_buffers(lhs, rhs, out):
236 linalg.elemwise_unary(lhs, outs=[out], emit_generic=True)
237 linalg.elemwise_binary(out, rhs, outs=[out], emit_generic=True)
244 def elemwise_log_mul_on_buffers(lhs, rhs, out):
246 lhs, outs=[out], fun=UnaryFn.log, emit_generic=True
249 out, rhs, outs=[out], fun=BinaryFn.mul, emit_generic=True
278 def fill_0d_on_buffers(value, out):
279 linalg.fill(value, outs=[out])
282 def fill_1d_on_buffers(value, out):
283 linalg.fill(value, outs=[out])
286 def fill_2d_on_buffers(value, out):
287 linalg.fill(value, outs=[out])
313 def fill_0d_on_buffers(value, out):
314 linalg.fill(value, outs=[out], emit_generic=True)
317 def fill_1d_on_buffers(value, out):
318 linalg.fill(value, outs=[out], emit_generic=True)
321 def fill_2d_on_buffers(value, out):
322 linalg.fill(value, outs=[out], emit_generic=True)
348 def fill_rng_on_buffers(min, max, seed, out):
349 linalg.fill_rng_2d(min, max, seed, outs=[out])
375 def fill_rng_on_buffers(min, max, seed, out):
376 linalg.fill_rng_2d(min, max, seed, outs=[out], emit_generic=True)