Uses of Class
org.bytedeco.javacpp.mkldnn.primitive.at

Packages that use mkldnn.primitive.at

Package	Description
org.bytedeco.javacpp

Uses of mkldnn.primitive.at in org.bytedeco.javacpp

Methods in org.bytedeco.javacpp that return mkldnn.primitive.at

Modifier and Type	Method and Description
mkldnn.primitive.at	mkldnn.primitive.at.data(mkldnn.mkldnn_primitive_at_t data)

Constructors in org.bytedeco.javacpp with parameters of type mkldnn.primitive.at

Constructor and Description
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at mean, mkldnn.primitive.at variance, mkldnn.primitive.at diff_dst, mkldnn.memory diff_src)
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at mean, mkldnn.primitive.at variance, mkldnn.primitive.at diff_dst, mkldnn.primitive.at weights_or_workspace, mkldnn.memory diff_src) \warning This constructor works for backward_data propagation - w/ weights but w/o workspace, or - w/ workspace but w/o weights
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at mean, mkldnn.primitive.at variance, mkldnn.primitive.at diff_dst, mkldnn.primitive.at weights, mkldnn.memory diff_src, mkldnn.memory diff_weights)
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at mean, mkldnn.primitive.at variance, mkldnn.primitive.at diff_dst, mkldnn.primitive.at weights, mkldnn.primitive.at workspace, mkldnn.memory diff_src, mkldnn.memory diff_weights)
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst)
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst, mkldnn.memory mean, mkldnn.memory variance)
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst, mkldnn.memory mean, mkldnn.memory variance, mkldnn.memory workspace) \warning batch_normalization_forward has 2 constructors with very similar signatures: - (pd, src, weights, dst, mean, variance) // 2 in, 3 out - (pd, src, dst, mean, variance, workspace) // 1 in, 4 out The only way to distinguish between those is to explicitly cast all input parameters to their type, i.e.
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst)
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst, mkldnn.memory mean, mkldnn.memory variance) \warning batch_normalization_forward has 2 constructors with very similar signatures: - (pd, src, weights, dst, mean, variance) // 2 in, 3 out - (pd, src, dst, mean, variance, workspace) // 1 in, 4 out The only way to distinguish between those is to explicitly cast all input parameters to their type, i.e.
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst, mkldnn.memory mean, mkldnn.memory variance, mkldnn.memory workspace)
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at mean, mkldnn.primitive.at variance, mkldnn.memory dst)
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at mean, mkldnn.primitive.at variance, mkldnn.primitive.at weights, mkldnn.memory dst)
concat(mkldnn.concat.primitive_desc concat_pd, mkldnn.primitive.at inputs, mkldnn.memory output)
convolution_backward_data(mkldnn.convolution_backward_data.primitive_desc aprimitive_desc, mkldnn.primitive.at diff_dst, mkldnn.primitive.at weights, mkldnn.memory diff_src)
convolution_backward_weights(mkldnn.convolution_backward_weights.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_weights)
convolution_backward_weights(mkldnn.convolution_backward_weights.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_weights, mkldnn.memory diff_bias)
convolution_forward(mkldnn.convolution_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst)
convolution_forward(mkldnn.convolution_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.primitive.at bias, mkldnn.memory dst)
convolution_relu_forward(mkldnn.convolution_relu_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst) Deprecated. consider using convolution_forward + post_ops
convolution_relu_forward(mkldnn.convolution_relu_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.primitive.at bias, mkldnn.memory dst) Deprecated. consider using convolution_forward + post_ops
deconvolution_backward_data(mkldnn.deconvolution_backward_data.primitive_desc aprimitive_desc, mkldnn.primitive.at diff_dst, mkldnn.primitive.at weights, mkldnn.memory diff_src)
deconvolution_backward_weights(mkldnn.deconvolution_backward_weights.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_weights)
deconvolution_backward_weights(mkldnn.deconvolution_backward_weights.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_weights, mkldnn.memory diff_bias)
deconvolution_forward(mkldnn.deconvolution_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst)
deconvolution_forward(mkldnn.deconvolution_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.primitive.at bias, mkldnn.memory dst)
eltwise_backward(mkldnn.eltwise_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_src)
eltwise_forward(mkldnn.eltwise_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst)
inner_product_backward_data(mkldnn.inner_product_backward_data.primitive_desc aprimitive_desc, mkldnn.primitive.at diff_dst, mkldnn.primitive.at weights, mkldnn.memory diff_src)
inner_product_backward_weights(mkldnn.inner_product_backward_weights.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_weights)
inner_product_backward_weights(mkldnn.inner_product_backward_weights.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_weights, mkldnn.memory diff_bias)
inner_product_forward(mkldnn.inner_product_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.memory dst)
inner_product_forward(mkldnn.inner_product_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at weights, mkldnn.primitive.at bias, mkldnn.memory dst)
lrn_backward(mkldnn.lrn_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.memory diff_src)
lrn_backward(mkldnn.lrn_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.primitive.at diff_dst, mkldnn.primitive.at workspace, mkldnn.memory diff_src)
lrn_forward(mkldnn.lrn_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst)
lrn_forward(mkldnn.lrn_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory workspace, mkldnn.memory dst)
pooling_backward(mkldnn.pooling_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at diff_dst, mkldnn.memory diff_src)
pooling_backward(mkldnn.pooling_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at diff_dst, mkldnn.primitive.at workspace, mkldnn.memory diff_src)
pooling_forward(mkldnn.pooling_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst)
pooling_forward(mkldnn.pooling_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst, mkldnn.memory workspace)
reorder(mkldnn.primitive.at input, mkldnn.memory output)
reorder(mkldnn.reorder.primitive_desc aprimitive_desc, mkldnn.primitive.at input, mkldnn.memory output)
rnn_backward(mkldnn.rnn_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at src_layer, mkldnn.primitive.at src_iter, mkldnn.primitive.at weights_layer, mkldnn.primitive.at weights_iter, mkldnn.primitive.at bias, mkldnn.primitive.at dst_layer, mkldnn.primitive.at dst_iter, mkldnn.memory diff_src_layer, mkldnn.memory diff_src_iter, mkldnn.memory diff_weights_layer, mkldnn.memory diff_weights_iter, mkldnn.memory diff_bias, mkldnn.primitive.at diff_dst_layer, mkldnn.primitive.at diff_dst_iter, mkldnn.primitive.at workspace)
rnn_forward(mkldnn.rnn_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src_layer, mkldnn.primitive.at src_iter, mkldnn.primitive.at weights_layer, mkldnn.primitive.at weights_iter, mkldnn.primitive.at bias, mkldnn.memory dst_layer, mkldnn.memory dst_iter, mkldnn.memory workspace)
softmax_backward(mkldnn.softmax_backward.primitive_desc aprimitive_desc, mkldnn.primitive.at dst, mkldnn.primitive.at diff_dst, mkldnn.memory diff_src)
softmax_forward(mkldnn.softmax_forward.primitive_desc aprimitive_desc, mkldnn.primitive.at src, mkldnn.memory dst)
sum(mkldnn.sum.primitive_desc sum_pd, mkldnn.primitive.at inputs, mkldnn.memory output)
view(mkldnn.view.primitive_desc view_pd, mkldnn.primitive.at input)