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

Packages that use primitive.at

Package	Description
org.bytedeco.mkldnn

Uses of primitive.at in org.bytedeco.mkldnn

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

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

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

Constructor and Description
batch_normalization_backward(batch_normalization_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at mean, primitive.at variance, primitive.at diff_dst, memory diff_src)
batch_normalization_backward(batch_normalization_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at mean, primitive.at variance, primitive.at diff_dst, primitive.at weights_or_workspace, 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(batch_normalization_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at mean, primitive.at variance, primitive.at diff_dst, primitive.at weights, memory diff_src, memory diff_weights)
batch_normalization_backward(batch_normalization_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at mean, primitive.at variance, primitive.at diff_dst, primitive.at weights, primitive.at workspace, memory diff_src, memory diff_weights)
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst)
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst, memory mean, memory variance)
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst, memory mean, memory variance, memory workspace) \warning batch_normalization_forward has two 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 them is to explicitly cast all input parameters to their type; that is, to const primitive:at &.
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, memory dst)
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, memory dst, memory mean, memory variance) \warning batch_normalization_forward has two 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 them is to explicitly cast all input parameters to their type; that is, to const primitive:at &.
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, memory dst, memory mean, memory variance, memory workspace)
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at mean, primitive.at variance, memory dst)
batch_normalization_forward(batch_normalization_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at mean, primitive.at variance, primitive.at weights, memory dst)
concat(concat.primitive_desc concat_pd, primitive.at inputs, memory output)
convolution_backward_data(convolution_backward_data.primitive_desc aprimitive_desc, primitive.at diff_dst, primitive.at weights, memory diff_src)
convolution_backward_weights(convolution_backward_weights.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_weights)
convolution_backward_weights(convolution_backward_weights.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_weights, memory diff_bias)
convolution_forward(convolution_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, memory dst)
convolution_forward(convolution_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, primitive.at bias, memory dst)
deconvolution_backward_data(deconvolution_backward_data.primitive_desc aprimitive_desc, primitive.at diff_dst, primitive.at weights, memory diff_src)
deconvolution_backward_weights(deconvolution_backward_weights.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_weights)
deconvolution_backward_weights(deconvolution_backward_weights.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_weights, memory diff_bias)
deconvolution_forward(deconvolution_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, memory dst)
deconvolution_forward(deconvolution_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, primitive.at bias, memory dst)
eltwise_backward(eltwise_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_src)
eltwise_forward(eltwise_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst)
inner_product_backward_data(inner_product_backward_data.primitive_desc aprimitive_desc, primitive.at diff_dst, primitive.at weights, memory diff_src)
inner_product_backward_weights(inner_product_backward_weights.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_weights)
inner_product_backward_weights(inner_product_backward_weights.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_weights, memory diff_bias)
inner_product_forward(inner_product_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, memory dst)
inner_product_forward(inner_product_forward.primitive_desc aprimitive_desc, primitive.at src, primitive.at weights, primitive.at bias, memory dst)
lrn_backward(lrn_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, memory diff_src)
lrn_backward(lrn_backward.primitive_desc aprimitive_desc, primitive.at src, primitive.at diff_dst, primitive.at workspace, memory diff_src)
lrn_forward(lrn_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst)
lrn_forward(lrn_forward.primitive_desc aprimitive_desc, primitive.at src, memory workspace, memory dst)
pooling_backward(pooling_backward.primitive_desc aprimitive_desc, primitive.at diff_dst, memory diff_src)
pooling_backward(pooling_backward.primitive_desc aprimitive_desc, primitive.at diff_dst, primitive.at workspace, memory diff_src)
pooling_forward(pooling_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst)
pooling_forward(pooling_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst, memory workspace)
reorder(primitive.at input, memory output)
reorder(reorder.primitive_desc aprimitive_desc, primitive.at input, memory output)
rnn_backward(rnn_backward.primitive_desc aprimitive_desc, primitive.at src_layer, primitive.at src_iter, primitive.at weights_layer, primitive.at weights_iter, primitive.at bias, primitive.at dst_layer, primitive.at dst_iter, memory diff_src_layer, memory diff_src_iter, memory diff_weights_layer, memory diff_weights_iter, memory diff_bias, primitive.at diff_dst_layer, primitive.at diff_dst_iter, primitive.at workspace)
rnn_forward(rnn_forward.primitive_desc aprimitive_desc, primitive.at src_layer, primitive.at src_iter, primitive.at weights_layer, primitive.at weights_iter, primitive.at bias, memory dst_layer, memory dst_iter, memory workspace)
shuffle_backward(shuffle_backward.primitive_desc aprimitive_desc, primitive.at diff_dst, memory diff_src)
shuffle_forward(shuffle_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst)
softmax_backward(softmax_backward.primitive_desc aprimitive_desc, primitive.at dst, primitive.at diff_dst, memory diff_src)
softmax_forward(softmax_forward.primitive_desc aprimitive_desc, primitive.at src, memory dst)
sum(sum.primitive_desc sum_pd, primitive.at inputs, memory output)
view(view.primitive_desc view_pd, primitive.at input)