Home | History | Annotate | Download | only in cuda

Lines Matching refs:opaque

1010                                              dropout, state_memory.opaque(),
1610 input_desc.handles() /*xDesc*/, input_data.opaque() /*x*/,
1611 input_h_desc.handle() /*hxDesc*/, input_h_data.opaque() /*hx*/,
1612 input_c_desc.handle() /*cxDesc*/, input_c_data.opaque() /*cx*/,
1613 rnn_desc.params_handle() /*wDesc*/, params.opaque() /*w*/,
1614 output_desc.handles() /*yDesc*/, output_data->opaque() /*y*/,
1615 output_h_desc.handle() /*hyDesc*/, output_h_data->opaque() /*hy*/,
1616 output_c_desc.handle() /*cyDesc*/, output_c_data->opaque() /*cy*/,
1617 workspace.opaque() /*workspace*/,
1628 input_desc.handles() /*xDesc*/, input_data.opaque() /*x*/,
1629 input_h_desc.handle() /*hxDesc*/, input_h_data.opaque() /*hx*/,
1630 input_c_desc.handle() /*cxDesc*/, input_c_data.opaque() /*cx*/,
1631 rnn_desc.params_handle() /*wDesc*/, params.opaque() /*w*/,
1632 output_desc.handles() /*yDesc*/, output_data->opaque() /*y*/,
1633 output_h_desc.handle() /*hyDesc*/, output_h_data->opaque() /*hy*/,
1634 output_c_desc.handle() /*cyDesc*/, output_c_data->opaque() /*cy*/,
1635 workspace.opaque() /*workspace*/,
1637 reserve_space.opaque() /*reserveSpace*/,
1705 output_data.opaque() /*y*/, output_desc.handles() /*dyDesc*/,
1706 opaque() /*dy*/, output_h_desc.handle() /*dhyDesc*/,
1707 output_h_backprop_data.opaque() /*dhy*/,
1709 output_c_backprop_data.opaque() /*dcy*/,
1710 rnn_desc.params_handle() /*wDesc*/, params.opaque() /*w*/,
1711 input_h_desc.handle() /*hxDesc*/, input_h_data.opaque() /*hx*/,
1712 input_c_desc.handle() /*cxDesc*/, input_c_data.opaque() /*cx*/,
1713 input_desc.handles() /*dxDesc*/, input_backprop_data->opaque() /*dx*/,
1715 input_h_backprop_data->opaque() /*dhx*/,
1717 input_c_backprop_data->opaque() /*dcx*/, workspace.opaque() /*workspace*/,
1719 reserve_space_data->opaque() /*reserveSpace*/,
1733 input_desc.handles() /*xDesc*/, input_data.opaque() /*x*/,
1734 input_h_desc.handle() /*hxDesc*/, input_h_data.opaque() /*hx*/,
1735 output_desc.handles() /*yDesc*/, output_data.opaque() /*y*/,
1736 workspace.opaque() /*workspace*/,
1739 params_backprop_data->opaque() /*dw*/,
1740 reserve_space_data->opaque() /*reserveSpace*/,
2468 /*srcData=*/input_data.opaque(), /*filterDesc=*/filter.handle(),
2469 /*filterData=*/filter_data.opaque(), /*convDesc=*/conv.handle(),
2470 /*algo=*/algo, /*workSpace=*/scratch.opaque(),
2472 /*destDesc=*/output_nd.handle(), /*destData=*/output_data->opaque());
2581 auto side_input_data_ptr = (side_input_scale == 0) ? output_data->opaque()
2582 : side_input_data.opaque();
2586 << "\nconv_input_data.opaque() = " << conv_input_data.opaque()
2588 << "\nfilter_data.opaque() = " << filter_data.opaque()
2590 << "\nscratch.opaque() = " << scratch.opaque()
2596 << "\nbiases.opaque() = " << biases.opaque()
2599 << "\noutput_data->opaque() = " << output_data->opaque();
2603 /*srcDesc=*/conv_input_nd.handle(), /*srcData=*/conv_input_data.opaque(),
2604 /*filterDesc=*/filter.handle(), /*filterData=*/filter_data.opaque(),
2605 /*convDesc=*/conv.handle(), algo, /*workSpace=*/scratch.opaque(),
2608 /*biasDesc=*/bias_nd.handle(), /*bias=*/biases.opaque(),
2610 /*destDesc=*/output_nd.handle(), /*destData=*/output_data->opaque());
2812 batch_mean_opaque = batch_mean->opaque();
2813 opaque();
2821 x_descriptor.handle(), x.opaque(), x_descriptor.handle(), y->opaque(),
2822 scale_offset_descriptor.handle(), scale.opaque(), offset.opaque(), 1.0,
2823 batch_mean_opaque, batch_var_opaque, epsilon, saved_mean->opaque(),
2824 saved_inv_var->opaque());
2832 const void* maybe_inv_var = var_to_inv_var().opaque();
2834 const void* maybe_inv_var = estimated_variance.opaque();
2838 x_descriptor.handle(), x.opaque(), x_descriptor.handle(), y->opaque(),
2839 scale_offset_descriptor.handle(), scale.opaque(), offset.opaque(),
2840 estimated_mean.opaque(), maybe_inv_var, epsilon);
2911 x_descriptor.handle(), x.opaque(), x_descriptor.handle(),
2912 y_backprop.opaque(), x_descriptor.handle(), x_backprop->opaque(),
2913 scale_offset_descriptor.handle(), scale.opaque(),
2914 scale_backprop->opaque(), offset_backprop->opaque(), epsilon,
2915 mean.opaque(), inv_var.opaque());
3097 backward_output_data.opaque(), &beta, transformed_out_back_nd.handle(),
3098 (*transform_scratch)->mutable_device_memory()->opaque());
3122 input_data.opaque(), &beta, output_tensor_desc.handle(),
3123 output_data->opaque());
3315 /*filterData=*/filter_data.opaque(),
3317 /*diffData=*/backward_output_data.opaque(),
3320 /*workSpace=*/scratch.opaque(),
3324 /*gradData=*/backward_input_data->opaque());
3562 /*srcData=*/input_data.opaque(),
3564 /*diffData=*/backward_output_data.opaque(),
3567 /*workSpace=*/scratch.opaque(),
3571 /*gradData=*/backward_filter_data->opaque());
3655 input_data.opaque(), &beta, bias_nd.handle(),
3656 backward_bias_data->opaque());
3796 const_cast<float*>(reinterpret_cast<const float*>(weights.opaque())) +
3805 reinterpret_cast<const float*>(output_data->opaque())) +
3847 if (input_data.opaque() != output_data->opaque()) {
3875 biases.opaque(), &beta, input_descriptor.handle(), output_data->opaque());
3943 opaque(), &beta, input_nd.handle(),
3944 output_data->opaque());
3979 src_desc.handle(), input_data.opaque(), &beta, dest_desc.handle(),
3980 output_data->opaque());
4014 src_desc.handle(), input_data.opaque(), &beta, dest_desc.handle(),
4015 output_data->opaque());
4048 src_desc.handle(), input_data.opaque(), &beta, dest_desc.handle(),
4049 output_data->opaque());
4085 dest_desc.handle(), output_data.opaque(), dest_desc.handle(),
4086 input_diff_data.opaque(), src_desc.handle(), input_data.opaque(), &beta,
4087 src_desc.handle(), output_diff_data->opaque());
4123 dest_desc.handle(), output_data.opaque(), dest_desc.handle(),
4124 input_diff_data.opaque(), src_desc.handle(), input_data.opaque(), &beta,
4125 src_desc.handle(), output_diff_data->opaque());
4160 dest_desc.handle(), output_data.opaque(), dest_desc.handle(),
4161 input_diff_data.opaque(), src_desc.handle(), input_data.opaque(), &beta,
4162 src_desc.handle(), output_diff_data->opaque());
4211 CUDNN_LRN_CROSS_CHANNEL_DIM1, &alpha, dims.handle(), input_data.opaque(),
4212 &beta, dims.handle(), output_data->opaque());
4253 normalized_data.opaque(), dims.handle(),
4254 normalized_variable_gradient.opaque(), dims.handle(), raw_data.opaque(),
4255 &beta, dims.handle(), raw_variable_gradient->opaque());