Home
last modified time | relevance | path

Searched refs:output_desc (Results 1 – 20 of 20) sorted by relevance

/external/mesa3d/src/gallium/auxiliary/translate/
Dtranslate_sse.c465 const struct util_format_description *output_desc = in translate_attr_convert() local
482 if (input_desc->colorspace != output_desc->colorspace) in translate_attr_convert()
492 for (i = 1; i < output_desc->nr_channels; ++i) { in translate_attr_convert()
494 (&output_desc->channel[i], &output_desc->channel[0], in translate_attr_convert()
495 sizeof(output_desc->channel[0]))) { in translate_attr_convert()
500 for (i = 0; i < output_desc->nr_channels; ++i) { in translate_attr_convert()
501 if (output_desc->swizzle[i] < 4) in translate_attr_convert()
502 swizzle[output_desc->swizzle[i]] = input_desc->swizzle[i]; in translate_attr_convert()
512 for (i = 0; i < output_desc->nr_channels; ++i) { in translate_attr_convert()
518 for (i = 0; i < output_desc->nr_channels; ++i) { in translate_attr_convert()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc77 se::dnn::BatchDescriptor output_desc(3); in Compute() local
78 output_desc.set_count(in_batch) in Compute()
88 output_desc.set_spatial_dim(dim_i, in Compute()
103 output_desc, &output_data) in Compute()
Dpooling_ops_common.cc245 se::dnn::BatchDescriptor output_desc; in Compute() local
246 output_desc.set_count(batch_size) in Compute()
267 output_desc, &output_data) in Compute()
Dconv_ops_3d.cc358 se::dnn::BatchDescriptor output_desc(3); in launch() local
359 output_desc.set_count(in_batch) in launch()
460 output_desc, &output_ptr, &scratch_allocator, in launch()
488 filter_ptr, conv_desc, output_desc, in launch()
Dcudnn_rnn_ops.cc654 std::unique_ptr<RnnSequenceTensorDescriptor>* output_desc, in CreateForwardAndBackwardIODescriptors() argument
708 *output_desc = output_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
714 *output_desc = output_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
721 *output_desc = output_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
743 std::unique_ptr<RnnSequenceTensorDescriptor> output_desc; in DoForward() local
751 context, model_shapes, &input_desc, &state_desc, &output_desc, in DoForward()
774 *output_desc, &output_data, *state_desc, in DoForward()
805 std::unique_ptr<RnnSequenceTensorDescriptor> output_desc; in DoBackward() local
813 context, model_shapes, &input_desc, &state_desc, &output_desc, in DoBackward()
852 params_data, *output_desc, output_data, *state_desc, in DoBackward()
Dconv_grad_ops_3d.cc1261 se::dnn::BatchDescriptor output_desc(3); in Compute() local
1262 output_desc.set_count(dims.batch_size) in Compute()
1376 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in Compute()
1412 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in Compute()
1660 se::dnn::BatchDescriptor output_desc(3); in Compute() local
1661 output_desc.set_count(dims.batch_size) in Compute()
1782 input_desc, input_ptr, output_desc, out_backprop_ptr, in Compute()
1819 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
Dconv_ops.cc769 se::dnn::BatchDescriptor output_desc; in operator ()() local
770 output_desc.set_count(out_batch) in operator ()()
870 output_desc, &output_ptr, &scratch_allocator, in operator ()()
898 filter_ptr, conv_desc, output_desc, in operator ()()
Dconv_grad_filter_ops.cc722 se::dnn::BatchDescriptor output_desc; in operator ()() local
723 output_desc.set_count(dims.batch_size) in operator ()()
856 input_desc, input_ptr, output_desc, out_backprop_ptr, in operator ()()
887 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
Dconv_grad_input_ops.cc838 se::dnn::BatchDescriptor output_desc; in operator ()() local
839 output_desc.set_count(dims.batch_size) in operator ()()
968 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in operator ()()
997 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
Dconv_ops_fused_impl.h703 se::dnn::BatchDescriptor output_desc;
704 output_desc.set_count(out_batch)
788 output_desc, &output_ptr, // output
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h81 const dnn::RnnSequenceTensorDescriptor& output_desc,
99 const dnn::RnnSequenceTensorDescriptor& output_desc,
117 const dnn::RnnSequenceTensorDescriptor& output_desc,
135 const dnn::RnnSequenceTensorDescriptor& output_desc,
160 const dnn::RnnSequenceTensorDescriptor& output_desc,
185 const dnn::RnnSequenceTensorDescriptor& output_desc,
546 const dnn::BatchDescriptor& output_desc,
614 const CudnnRnnSequenceTensorDescriptor& output_desc,
633 const CudnnRnnSequenceTensorDescriptor& output_desc,
Dcuda_dnn.cc1419 const CudnnRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward() argument
1447 if (!(output_desc.max_seq_length() == model_dims.max_seq_length && in ExtractAndCheckRnnForward()
1448 output_desc.batch_size() == model_dims.batch_size && in ExtractAndCheckRnnForward()
1449 output_desc.data_size() == in ExtractAndCheckRnnForward()
1514 const CudnnRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl() argument
1527 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
1575 /*yDesc=*/output_desc.data_handle(), in DoRnnForwardImpl()
1596 /*w=*/params.opaque(), /*yDesc=*/output_desc.handles(), in DoRnnForwardImpl()
1612 /*yDesc=*/output_desc.data_handle(), in DoRnnForwardImpl()
1635 /*w=*/params.opaque(), /*yDesc=*/output_desc.handles(), in DoRnnForwardImpl()
[all …]
/external/curl/lib/
Dsocks_sspi.c77 SecBufferDesc input_desc, output_desc, wrap_desc; in Curl_SOCKS5_gssapi_negotiate() local
124 output_desc.cBuffers = 1; in Curl_SOCKS5_gssapi_negotiate()
125 output_desc.pBuffers = &sspi_send_token; in Curl_SOCKS5_gssapi_negotiate()
126 output_desc.ulVersion = SECBUFFER_VERSION; in Curl_SOCKS5_gssapi_negotiate()
177 &output_desc, in Curl_SOCKS5_gssapi_negotiate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h72 const dnn::RnnSequenceTensorDescriptor& output_desc,
90 const dnn::RnnSequenceTensorDescriptor& output_desc,
108 const dnn::RnnSequenceTensorDescriptor& output_desc,
126 const dnn::RnnSequenceTensorDescriptor& output_desc,
151 const dnn::RnnSequenceTensorDescriptor& output_desc,
176 const dnn::RnnSequenceTensorDescriptor& output_desc,
529 const dnn::BatchDescriptor& output_desc,
671 const MIOpenRnnSequenceTensorDescriptor& output_desc,
688 const MIOpenRnnSequenceTensorDescriptor& output_desc,
Drocm_dnn.cc1899 const MIOpenRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward() argument
1928 if (!(output_desc.seq_length() == model_dims->seq_length && in ExtractAndCheckRnnForward()
1929 output_desc.batch_size() == model_dims->batch_size && in ExtractAndCheckRnnForward()
1930 output_desc.data_size() == in ExtractAndCheckRnnForward()
2009 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl() argument
2021 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2081 params.opaque() /*w*/, output_desc.handles() /*yDesc*/, in DoRnnForwardImpl()
2099 params.opaque() /*w*/, output_desc.handles() /*yDesc*/, in DoRnnForwardImpl()
2124 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnBackwardImpl() argument
2143 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc605 dnn::BatchDescriptor output_desc; in launch() local
606 output_desc.set_count(batch_size) in launch()
726 output_desc, &output_ptr, &scratch_allocator, in launch()
759 bias_desc, bias_ptr, dnn_activation_mode, output_desc, in launch()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2124 const dnn::RnnSequenceTensorDescriptor& output_desc,
2145 const dnn::RnnSequenceTensorDescriptor& output_desc,
2166 const dnn::RnnSequenceTensorDescriptor& output_desc,
2228 const dnn::RnnSequenceTensorDescriptor& output_desc,
2256 const dnn::RnnSequenceTensorDescriptor& output_desc,
2284 const dnn::RnnSequenceTensorDescriptor& output_desc,
2319 const dnn::BatchDescriptor& output_desc,
Dstream.h1745 const dnn::RnnSequenceTensorDescriptor &output_desc,
1764 const dnn::RnnSequenceTensorDescriptor &output_desc,
1782 const dnn::RnnSequenceTensorDescriptor &output_desc,
1803 const dnn::RnnSequenceTensorDescriptor &output_desc,
1828 const dnn::RnnSequenceTensorDescriptor &output_desc,
1853 const dnn::RnnSequenceTensorDescriptor &output_desc,
1875 const dnn::BatchDescriptor &output_desc,
1884 const dnn::BatchDescriptor &output_desc, in ThenTransformTensor() argument
1887 input_data, output_desc, in ThenTransformTensor()
Dstream.cc4854 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward() argument
4868 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4890 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward() argument
4904 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4927 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward() argument
4941 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4964 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward() argument
4985 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
5010 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward() argument
5031 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
[all …]
/external/libxcam/modules/ocl/
Dcl_pyramid_blender.cpp1100 CLImageDesc input_desc, output_desc; in prepare_arguments() local
1102 output_desc.format.image_channel_data_type = CL_UNSIGNED_INT8; in prepare_arguments()
1103 output_desc.format.image_channel_order = CL_RGBA; in prepare_arguments()
1104 output_desc.width = input_desc.width * 2; in prepare_arguments()
1105 output_desc.height = input_desc.height; in prepare_arguments()
1106 output_desc.row_pitch = input_desc.row_pitch; in prepare_arguments()
1109 change_image_format (context, out_orig, output_scale_image, output_desc); in prepare_arguments()