Home
last modified time | relevance | path

Searched refs:input_desc (Results 1 – 21 of 21) sorted by relevance

/external/mesa3d/src/gallium/auxiliary/translate/
Dtranslate_sse.c463 const struct util_format_description *input_desc = in translate_attr_convert() local
479 if (input_desc->channel[0].size & 7) in translate_attr_convert()
482 if (input_desc->colorspace != output_desc->colorspace) in translate_attr_convert()
485 for (i = 1; i < input_desc->nr_channels; ++i) { in translate_attr_convert()
487 (&input_desc->channel[i], &input_desc->channel[0], in translate_attr_convert()
488 sizeof(input_desc->channel[0]))) in translate_attr_convert()
502 swizzle[output_desc->swizzle[i]] = input_desc->swizzle[i]; in translate_attr_convert()
514 && i >= input_desc->nr_channels) in translate_attr_convert()
526 switch (input_desc->channel[0].type) { in translate_attr_convert()
531 input_desc->channel[0].size * in translate_attr_convert()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1161 TensorDescriptor input_desc = CreateTensorDescriptor(); in Create() local
1165 /*tensorDesc=*/input_desc.get(), /*dataType=*/data_type, in Create()
1173 /*xDesc=*/input_desc.get(), /*sizeInBytes=*/&params_size, in Create()
1215 /*layer=*/layer, /*xDesc=*/input_desc.get(), in Create()
1222 /*layer=*/layer, /*xDesc=*/input_desc.get(), in Create()
1413 const CudnnRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward() argument
1428 model_dims.batch_size = input_desc.batch_size(); in ExtractAndCheckRnnForward()
1429 model_dims.max_seq_length = input_desc.max_seq_length(); in ExtractAndCheckRnnForward()
1431 model_dims.input_size = input_desc.data_size(); in ExtractAndCheckRnnForward()
1471 const CudnnRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize() argument
[all …]
Dcuda_dnn.h74 const dnn::RnnSequenceTensorDescriptor& input_desc,
92 const dnn::RnnSequenceTensorDescriptor& input_desc,
110 const dnn::RnnSequenceTensorDescriptor& input_desc,
128 const dnn::RnnSequenceTensorDescriptor& input_desc,
153 const dnn::RnnSequenceTensorDescriptor& input_desc,
178 const dnn::RnnSequenceTensorDescriptor& input_desc,
543 bool DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc,
608 const CudnnRnnSequenceTensorDescriptor& input_desc,
627 const CudnnRnnSequenceTensorDescriptor& input_desc,
/external/curl/lib/vauth/
Dkrb5_sspi.c279 SecBufferDesc input_desc; in Curl_auth_create_gssapi_security_message() local
326 input_desc.ulVersion = SECBUFFER_VERSION; in Curl_auth_create_gssapi_security_message()
327 input_desc.cBuffers = 2; in Curl_auth_create_gssapi_security_message()
328 input_desc.pBuffers = input_buf; in Curl_auth_create_gssapi_security_message()
337 status = s_pSecFn->DecryptMessage(krb5->context, &input_desc, 0, &qop); in Curl_auth_create_gssapi_security_message()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1893 const MIOpenRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward() argument
1907 model_dims->batch_size = input_desc.batch_size(); in ExtractAndCheckRnnForward()
1908 model_dims->seq_length = input_desc.seq_length(); in ExtractAndCheckRnnForward()
1910 model_dims->input_size = input_desc.data_size(); in ExtractAndCheckRnnForward()
1953 const MIOpenRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize() argument
1957 input_desc.handles()[0] /*xDesc*/, &params_size_in_bytes /*sizeInBytes*/, in CheckRNNParameterSize()
1969 const MIOpenRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace() argument
1976 input_desc.seq_length() /*seqLength*/, input_desc.handles() /*xDesc*/, in CreateRnnWorkspace()
2003 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl() argument
2020 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnForwardImpl()
[all …]
Drocm_dnn.h65 const dnn::RnnSequenceTensorDescriptor& input_desc,
83 const dnn::RnnSequenceTensorDescriptor& input_desc,
101 const dnn::RnnSequenceTensorDescriptor& input_desc,
119 const dnn::RnnSequenceTensorDescriptor& input_desc,
144 const dnn::RnnSequenceTensorDescriptor& input_desc,
169 const dnn::RnnSequenceTensorDescriptor& input_desc,
526 bool DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc,
664 const MIOpenRnnSequenceTensorDescriptor& input_desc,
681 const MIOpenRnnSequenceTensorDescriptor& input_desc,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc73 se::dnn::BatchDescriptor input_desc(3); in Compute() local
74 input_desc.set_count(in_batch) in Compute()
86 input_desc.set_spatial_dim(dim_i, in Compute()
102 ->ThenPoolForward(pooling_desc, input_desc, input_data, in Compute()
Dpooling_ops_common.cc238 se::dnn::BatchDescriptor input_desc; in Compute() local
239 input_desc.set_count(batch_size) in Compute()
266 ->ThenPoolForward(pooling_desc, input_desc, input_data, in Compute()
Dconv_ops_3d.cc351 se::dnn::BatchDescriptor input_desc(3); in launch() local
352 input_desc.set_count(in_batch) in launch()
459 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
487 ->ThenConvolveWithAlgorithm(input_desc, input_ptr, filter_desc, in launch()
Dcudnn_rnn_ops.cc652 std::unique_ptr<RnnSequenceTensorDescriptor>* input_desc, in CreateForwardAndBackwardIODescriptors() argument
670 *input_desc = input_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
676 *input_desc = input_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
683 *input_desc = input_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
741 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoForward() local
751 context, model_shapes, &input_desc, &state_desc, &output_desc, in DoForward()
772 ->ThenRnnForward(rnn_desc, *input_desc, input_data, *state_desc, in DoForward()
803 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoBackward() local
813 context, model_shapes, &input_desc, &state_desc, &output_desc, in DoBackward()
850 ->ThenRnnBackward(rnn_desc, *input_desc, input_data, *state_desc, in DoBackward()
Dconv_grad_filter_ops.cc716 se::dnn::BatchDescriptor input_desc; in operator ()() local
717 input_desc.set_count(dims.batch_size) in operator ()()
823 {{input_desc.height(), // in_rows in operator ()()
824 input_desc.width()}}, // in_cols 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.cc832 se::dnn::BatchDescriptor input_desc; in operator ()() local
833 input_desc.set_count(dims.batch_size) in operator ()()
935 {{input_desc.height(), // in_rows in operator ()()
936 input_desc.width()}}, // in_cols in operator ()()
969 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
998 input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
Dconv_grad_ops_3d.cc1254 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1255 input_desc.set_count(dims.batch_size) in Compute()
1377 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1413 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1650 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1651 input_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.cc763 se::dnn::BatchDescriptor input_desc; in operator ()() local
764 input_desc.set_count(in_batch) in operator ()()
869 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
897 ->ThenConvolveWithAlgorithm(input_desc, input_ptr, filter_desc, in operator ()()
Dconv_ops_fused_impl.h678 se::dnn::BatchDescriptor input_desc;
679 input_desc.set_count(in_batch)
781 input_desc, input_ptr, // input
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc61 dnn::BatchDescriptor input_desc; in MakeDescriptors() local
62 input_desc.set_layout(dnn::DataLayout::kBatchDepthYX) in MakeDescriptors()
70 .set_feature_map_count(input_desc.feature_map_count()) in MakeDescriptors()
75 return std::make_pair(input_desc, scale_offset_desc); in MakeDescriptors()
/external/curl/lib/
Dsocks_sspi.c77 SecBufferDesc input_desc, output_desc, wrap_desc; in Curl_SOCKS5_gssapi_negotiate() local
116 input_desc.cBuffers = 1; in Curl_SOCKS5_gssapi_negotiate()
117 input_desc.pBuffers = &sspi_recv_token; in Curl_SOCKS5_gssapi_negotiate()
118 input_desc.ulVersion = SECBUFFER_VERSION; in Curl_SOCKS5_gssapi_negotiate()
174 &input_desc, in Curl_SOCKS5_gssapi_negotiate()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2117 const dnn::RnnSequenceTensorDescriptor& input_desc,
2138 const dnn::RnnSequenceTensorDescriptor& input_desc,
2159 const dnn::RnnSequenceTensorDescriptor& input_desc,
2221 const dnn::RnnSequenceTensorDescriptor& input_desc,
2249 const dnn::RnnSequenceTensorDescriptor& input_desc,
2277 const dnn::RnnSequenceTensorDescriptor& input_desc,
2316 const dnn::BatchDescriptor& input_desc,
Dstream.h1738 const dnn::RnnSequenceTensorDescriptor &input_desc,
1757 const dnn::RnnSequenceTensorDescriptor &input_desc,
1775 const dnn::RnnSequenceTensorDescriptor &input_desc,
1796 const dnn::RnnSequenceTensorDescriptor &input_desc,
1821 const dnn::RnnSequenceTensorDescriptor &input_desc,
1846 const dnn::RnnSequenceTensorDescriptor &input_desc,
1872 Stream &ThenTransformTensor(const dnn::BatchDescriptor &input_desc,
1882 Stream &ThenTransformTensor(const dnn::BatchDescriptor &input_desc, in ThenTransformTensor() argument
1886 return ThenTransformTensor(input_desc, dnn::ToDataType<InElemT>(), in ThenTransformTensor()
Dstream.cc4847 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward() argument
4867 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4884 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward() argument
4903 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4920 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward() argument
4940 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4957 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward() argument
4984 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
5004 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward() argument
5030 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
[all …]
/external/libxcam/modules/ocl/
Dcl_pyramid_blender.cpp1100 CLImageDesc input_desc, output_desc; in prepare_arguments() local
1101 input_desc = out_orig->get_image_desc (); 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()