io.improbable.keanu.tensor.jvm

Class JVMTensorBroadcast

java.lang.Object

io.improbable.keanu.tensor.jvm.JVMTensorBroadcast

public class JVMTensorBroadcast
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
JVMTensorBroadcast()

Method Summary

Modifier and Type	Method and Description
static <T,B extends JVMBuffer.PrimitiveArrayWrapper<T,B>> void	broadcast(B buffer, long[] shape, long[] stride, B outputBuffer, long[] outputStride)
static <IN,OUT,INBUFFER extends JVMBuffer.PrimitiveArrayWrapper<IN,INBUFFER>,OUTBUFFER extends JVMBuffer.PrimitiveArrayWrapper<OUT,OUTBUFFER>> ResultWrapper<OUT,OUTBUFFER>	broadcastIfNeeded(JVMBuffer.ArrayWrapperFactory<OUT,OUTBUFFER> factory, INBUFFER leftBuffer, long[] leftShape, long[] leftStride, long leftBufferLength, INBUFFER rightBuffer, long[] rightShape, long[] rightStride, long rightBufferLength, java.util.function.BiFunction<IN,IN,OUT> op, boolean inPlace)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

JVMTensorBroadcast

public JVMTensorBroadcast()

Method Detail

broadcastIfNeeded

public static <IN,OUT,INBUFFER extends JVMBuffer.PrimitiveArrayWrapper<IN,INBUFFER>,OUTBUFFER extends JVMBuffer.PrimitiveArrayWrapper<OUT,OUTBUFFER>> ResultWrapper<OUT,OUTBUFFER> broadcastIfNeeded(JVMBuffer.ArrayWrapperFactory<OUT,OUTBUFFER> factory,
                                                                                                                                                                                                     INBUFFER leftBuffer,
                                                                                                                                                                                                     long[] leftShape,
                                                                                                                                                                                                     long[] leftStride,
                                                                                                                                                                                                     long leftBufferLength,
                                                                                                                                                                                                     INBUFFER rightBuffer,
                                                                                                                                                                                                     long[] rightShape,
                                                                                                                                                                                                     long[] rightStride,
                                                                                                                                                                                                     long rightBufferLength,
                                                                                                                                                                                                     java.util.function.BiFunction<IN,IN,OUT> op,
                                                                                                                                                                                                     boolean inPlace)

broadcast

public static <T,B extends JVMBuffer.PrimitiveArrayWrapper<T,B>> void broadcast(B buffer,
                                                                                long[] shape,
                                                                                long[] stride,
                                                                                B outputBuffer,
                                                                                long[] outputStride)