Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbroadcast_to.h24 void BroadcastImpl(const NdArrayDesc<N>& input_desc, const char* input_data, in BroadcastImpl() argument
32 input_data + SubscriptToIndex(input_desc, indexes) * type_size; in BroadcastImpl()
42 for (indexes[dim] = 0; indexes[dim] < input_desc.extents[dim]; in BroadcastImpl()
44 BroadcastImpl<N>(input_desc, input_data, output_desc, output_data, indexes, in BroadcastImpl()
50 if (input_desc.extents[dim] != output_desc.extents[dim]) { in BroadcastImpl()
66 NdArrayDesc<N> input_desc; in BroadcastTo() local
69 &input_desc); in BroadcastTo()
77 if (input_desc.extents[i] != output_desc.extents[i]) { in BroadcastTo()
92 BroadcastImpl<N>(input_desc, input_data, output_desc, output_data, indexes, 0, in BroadcastTo()
Dtranspose.h38 NdArrayDesc<N> input_desc; in TransposeImpl() local
41 &input_desc); in TransposeImpl()
58 TFLITE_DCHECK_EQ(input_desc.extents[extended_perm[k]], in TransposeImpl()
60 perm_input_desc.extents[k] = input_desc.extents[extended_perm[k]]; in TransposeImpl()
61 perm_input_desc.strides[k] = input_desc.strides[extended_perm[k]]; in TransposeImpl()
/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.cc1286 const TensorDescriptor& input_desc, const FilterDescriptor& filter_desc, in CheckAndFetchProjectionWeights() argument
1332 /*layer=*/layer, /*xDesc=*/input_desc.get(), in CheckAndFetchProjectionWeights()
1360 TensorDescriptor input_desc = CreateTensorDescriptor(); in Create() local
1364 /*tensorDesc=*/input_desc.get(), /*dataType=*/data_type, in Create()
1372 /*xDesc=*/input_desc.get(), /*sizeInBytes=*/&params_size, in Create()
1415 /*layer=*/layer, /*xDesc=*/input_desc.get(), in Create()
1422 /*layer=*/layer, /*xDesc=*/input_desc.get(), in Create()
1444 cudnn, rnn_desc, layer, input_desc, filter_desc, region_desc_handle, in Create()
1599 const CudnnRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward() argument
1614 model_dims.batch_size = input_desc.batch_size(); in ExtractAndCheckRnnForward()
[all …]
Dcuda_dnn.h75 const dnn::RnnSequenceTensorDescriptor& input_desc,
93 const dnn::RnnSequenceTensorDescriptor& input_desc,
111 const dnn::RnnSequenceTensorDescriptor& input_desc,
129 const dnn::RnnSequenceTensorDescriptor& input_desc,
154 const dnn::RnnSequenceTensorDescriptor& input_desc,
179 const dnn::RnnSequenceTensorDescriptor& input_desc,
574 bool DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc,
642 const CudnnRnnSequenceTensorDescriptor& input_desc,
661 const CudnnRnnSequenceTensorDescriptor& 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()
110 ->ThenPoolForward(pooling_desc, input_desc, input_data, output_desc, in Compute()
115 ->ThenPoolForward(pooling_desc, input_desc, input_data, in Compute()
Dconv_grad_input_ops.cc249 se::dnn::BatchDescriptor input_desc; in operator ()() local
250 input_desc.set_count(dims.batch_size) in operator ()()
365 {{input_desc.height(), // in_rows in operator ()()
366 input_desc.width()}}, // in_cols in operator ()()
422 input_desc, &in_backprop_ptr_rz, allocator_used, in operator ()()
448 se::dnn::ToDataType<T>::value, stream, input_desc, in_backprop_ptr, in operator ()()
475 input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
494 in_backprop_ptr, filter_ptr, out_backprop_ptr, input_desc, filter_desc, in operator ()()
502 input_desc, &in_backprop_ptr, &scratch_allocator, algorithm_config, in operator ()()
Dgpu_utils.cc131 const se::dnn::BatchDescriptor& input_desc, in LogConvAutotuneResults() argument
141 *instr.mutable_input() = input_desc.ToProto(element_type); in LogConvAutotuneResults()
174 const se::dnn::BatchDescriptor& input_desc, in LogFusedConvForwardAutotuneResults() argument
184 *instr.mutable_input() = input_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
Dconv_ops_3d.cc397 se::dnn::BatchDescriptor input_desc(3); in launch() local
398 input_desc.set_count(in_batch) in launch()
541 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
570 se::dnn::ToDataType<T>::value, stream, input_desc, in launch()
595 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
618 filter_ptr, output_ptr, input_desc, filter_desc, in launch()
626 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in launch()
Dconv_grad_filter_ops.cc854 se::dnn::BatchDescriptor input_desc; in operator ()() local
855 input_desc.set_count(dims.batch_size) in operator ()()
960 {{input_desc.height(), // in_rows in operator ()()
961 input_desc.width()}}, // in_cols in operator ()()
1016 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1046 se::dnn::ToDataType<T>::value, stream, input_desc, input_ptr, in operator ()()
1072 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1092 filter_backprop_ptr, out_backprop_ptr, input_desc, in operator ()()
1101 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
Dconv_grad_ops_3d.cc1298 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1299 input_desc.set_count(dims.batch_size) in Compute()
1456 input_desc, &in_backprop_ptr_rz, allocator_used, in Compute()
1485 se::dnn::ToDataType<T>::value, stream, input_desc, in_backprop_ptr, in Compute()
1494 input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1513 filter_ptr, out_backprop_ptr, input_desc, in Compute()
1525 input_desc, &in_backprop_ptr, &scratch_allocator, algorithm_config, in Compute()
1792 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1793 input_desc.set_count(dims.batch_size) in Compute()
1939 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
[all …]
Dgpu_utils.h222 const se::dnn::BatchDescriptor& input_desc,
234 const se::dnn::BatchDescriptor& input_desc,
Dconv_ops.cc879 se::dnn::BatchDescriptor input_desc; in operator ()() local
880 input_desc.set_count(in_batch) in operator ()()
1028 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
1058 stream, input_desc, input_ptr, filter_desc, filter_ptr, output_desc, in operator ()()
1084 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
1104 output_tensor, input_desc, filter_desc, output_desc, in operator ()()
1117 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in operator ()()
Dpooling_ops_common.cc362 se::dnn::BatchDescriptor input_desc; in Compute() local
363 input_desc.set_count(batch_size) in Compute()
398 ->ThenPoolForward(pooling_desc, input_desc, input_data, output_desc, in Compute()
403 ->ThenPoolForward(pooling_desc, input_desc, input_data, in Compute()
Dcudnn_rnn_ops.cc693 std::unique_ptr<RnnSequenceTensorDescriptor>* input_desc, in CreateForwardAndBackwardIODescriptors() argument
713 *input_desc = input_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
719 *input_desc = input_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
726 *input_desc = input_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
799 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoForward() local
810 context, model_shapes, &input_desc, &h_state_desc, &c_state_desc, in DoForward()
831 ->ThenRnnForward(rnn_desc, *input_desc, input_data, *h_state_desc, in DoForward()
862 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoBackward() local
873 context, model_shapes, &input_desc, &h_state_desc, &c_state_desc, in DoBackward()
911 rnn_desc, *input_desc, input_data, *h_state_desc, input_h_data, in DoBackward()
Dconv_ops_fused_impl.h572 se::dnn::BatchDescriptor input_desc;
573 input_desc.set_count(in_batch)
674 input_desc, input_ptr, // input
692 output_ptr, bias_ptr, side_input_ptr, input_desc, filter_desc,
/external/curl/lib/vauth/
Dkrb5_sspi.c289 SecBufferDesc input_desc; in Curl_auth_create_gssapi_security_message() local
342 input_desc.ulVersion = SECBUFFER_VERSION; in Curl_auth_create_gssapi_security_message()
343 input_desc.cBuffers = 2; in Curl_auth_create_gssapi_security_message()
344 input_desc.pBuffers = input_buf; in Curl_auth_create_gssapi_security_message()
353 status = s_pSecFn->DecryptMessage(krb5->context, &input_desc, 0, &qop); in Curl_auth_create_gssapi_security_message()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc66 se::dnn::BatchDescriptor input_desc; member
98 batch_descs.input_desc.set_layout(se::dnn::DataLayout::kBatchDepthYX) in MakeBatchNormDescriptors()
105 .set_feature_map_count(batch_descs.input_desc.feature_map_count()) in MakeBatchNormDescriptors()
125 params->operand_desc = batch_descs.input_desc; in AssignCommonParams()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1974 const MIOpenRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward() argument
1988 model_dims->batch_size = input_desc.batch_size(); in ExtractAndCheckRnnForward()
1989 model_dims->seq_length = input_desc.seq_length(); in ExtractAndCheckRnnForward()
1991 model_dims->input_size = input_desc.data_size(); in ExtractAndCheckRnnForward()
2034 const MIOpenRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize() argument
2038 input_desc.handles()[0] /*xDesc*/, &params_size_in_bytes /*sizeInBytes*/, in CheckRNNParameterSize()
2050 const MIOpenRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace() argument
2057 input_desc.seq_length() /*seqLength*/, input_desc.handles() /*xDesc*/, in CreateRnnWorkspace()
2084 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl() argument
2101 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnForwardImpl()
[all …]
Drocm_dnn.h102 const dnn::RnnSequenceTensorDescriptor& input_desc,
120 const dnn::RnnSequenceTensorDescriptor& input_desc,
138 const dnn::RnnSequenceTensorDescriptor& input_desc,
156 const dnn::RnnSequenceTensorDescriptor& input_desc,
181 const dnn::RnnSequenceTensorDescriptor& input_desc,
206 const dnn::RnnSequenceTensorDescriptor& input_desc,
578 bool DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc,
739 const MIOpenRnnSequenceTensorDescriptor& input_desc,
756 const MIOpenRnnSequenceTensorDescriptor& input_desc,
/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()
176 &input_desc, in Curl_SOCKS5_gssapi_negotiate()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc160 for (const auto& input_desc : input_descriptors_) { in GetNextInternal() local
161 int64 start = input_desc.interval * slice_number_; in GetNextInternal()
162 int64 end = std::min(start + input_desc.interval, in GetNextInternal()
163 input_desc.original_batch_dim); in GetNextInternal()
171 Tensor slice = input_desc.whole_tensor.Slice(start, end); in GetNextInternal()
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_op.h112 for (const auto& input_desc : input_descs) { in Compute() local
113 free(input_desc.descriptor); in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2186 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward() argument
2207 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward() argument
2228 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward() argument
2290 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward() argument
2318 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward() argument
2346 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward() argument
2451 const dnn::BatchDescriptor& input_desc, in DoTransformTensor() argument
Dstream.h1780 const dnn::RnnSequenceTensorDescriptor &input_desc,
1799 const dnn::RnnSequenceTensorDescriptor &input_desc,
1817 const dnn::RnnSequenceTensorDescriptor &input_desc,
1838 const dnn::RnnSequenceTensorDescriptor &input_desc,
1863 const dnn::RnnSequenceTensorDescriptor &input_desc,
1888 const dnn::RnnSequenceTensorDescriptor &input_desc,
1926 Stream &ThenTransformTensor(const dnn::BatchDescriptor &input_desc,
1936 Stream &ThenTransformTensor(const dnn::BatchDescriptor &input_desc, in ThenTransformTensor() argument
1940 return ThenTransformTensor(input_desc, dnn::ToDataType<InElemT>(), in ThenTransformTensor()

12