Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbroadcast_to.h25 const NdArrayDesc<N>& output_desc, char* output_data, in BroadcastImpl() argument
30 int copy_size = output_desc.strides[dim] * type_size; in BroadcastImpl()
34 output_data + SubscriptToIndex(output_desc, indexes) * type_size; in BroadcastImpl()
35 for (int i = 0; i < output_desc.extents[dim]; ++i, data_dst += copy_size) { 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()
51 int copy_size = output_desc.strides[dim] * type_size; in BroadcastImpl()
53 output_data + SubscriptToIndex(output_desc, indexes) * type_size; in BroadcastImpl()
55 for (int i = 1; i < output_desc.extents[dim]; ++i, data_dst += copy_size) { in BroadcastImpl()
67 NdArrayDesc<N> output_desc; in BroadcastTo() local
[all …]
Dsub.h82 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
85 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
99 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastSubSlow()
105 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
122 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
125 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
159 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastSubSlow()
162 NDOpsHelper<N>(output_desc, sub_func); in BroadcastSubSlow()
179 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
182 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
[all …]
Ddiv.h87 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local
91 &output_desc); in BroadcastDivSlow()
121 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastDivSlow()
124 NDOpsHelper<N>(output_desc, div_func); in BroadcastDivSlow()
149 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local
153 &output_desc); in BroadcastDivSlow()
164 output_data[SubscriptToIndex(output_desc, indexes)] = in BroadcastDivSlow()
170 NDOpsHelper<N>(output_desc, div_func); in BroadcastDivSlow()
Dmaximum_minimum.h46 NdArrayDesc<N> output_desc; in MaximumMinimumBroadcastSlow() local
50 &output_desc); in MaximumMinimumBroadcastSlow()
53 output_data[SubscriptToIndex(output_desc, indexes)] = in MaximumMinimumBroadcastSlow()
57 NDOpsHelper<N>(output_desc, maxmin_func); in MaximumMinimumBroadcastSlow()
Dtranspose.h39 NdArrayDesc<N> output_desc; in TransposeImpl() local
43 &output_desc); in TransposeImpl()
59 output_desc.extents[k]); in TransposeImpl()
66 output_data[SubscriptToIndex(output_desc, indexes)] = in TransposeImpl()
69 NDOpsHelper<N>(output_desc, tranpose_func); in TransposeImpl()
/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()
110 ->ThenPoolForward(pooling_desc, input_desc, input_data, output_desc, in Compute()
116 output_desc, &output_data) in Compute()
Dgpu_utils.cc133 const se::dnn::BatchDescriptor& output_desc, in LogConvAutotuneResults() argument
143 *instr.mutable_output() = output_desc.ToProto(element_type); in LogConvAutotuneResults()
176 const se::dnn::BatchDescriptor& output_desc, in LogFusedConvForwardAutotuneResults() argument
186 *instr.mutable_output() = output_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
Dconv_grad_input_ops.cc255 se::dnn::BatchDescriptor output_desc; in operator ()() local
256 output_desc.set_count(dims.batch_size) in operator ()()
421 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
449 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
474 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
495 output_desc, conv_desc, stream->parent(), results); in operator ()()
501 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
Dconv_ops_3d.cc404 se::dnn::BatchDescriptor output_desc(3); in launch() local
405 output_desc.set_count(in_batch) in launch()
542 output_desc, &output_ptr_rz, allocator_used, in launch()
571 input_ptr, filter_desc, filter_ptr, output_desc, in launch()
596 output_desc, &output_ptr, &scratch_allocator, in launch()
619 output_desc, conv_desc, stream->parent(), results); in launch()
626 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in launch()
Dconv_grad_ops_3d.cc1305 se::dnn::BatchDescriptor output_desc(3); in Compute() local
1306 output_desc.set_count(dims.batch_size) in Compute()
1455 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1486 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1493 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1514 filter_desc, output_desc, conv_desc, in Compute()
1524 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1802 se::dnn::BatchDescriptor output_desc(3); in Compute() local
1803 output_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.h224 const se::dnn::BatchDescriptor& output_desc,
236 const se::dnn::BatchDescriptor& output_desc,
Dconv_grad_filter_ops.cc860 se::dnn::BatchDescriptor output_desc; in operator ()() local
861 output_desc.set_count(dims.batch_size) in operator ()()
1016 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1047 filter_desc, filter_backprop_ptr, output_desc, out_backprop_ptr, in operator ()()
1072 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1093 filter_desc, output_desc, conv_desc, in operator ()()
1101 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
Dconv_ops.cc885 se::dnn::BatchDescriptor output_desc; in operator ()() local
886 output_desc.set_count(out_batch) in operator ()()
1029 output_desc, &output_tensor, allocator_used, in operator ()()
1058 stream, input_desc, input_ptr, filter_desc, filter_ptr, output_desc, in operator ()()
1085 output_desc, &output_ptr, &scratch_allocator, 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.cc369 se::dnn::BatchDescriptor output_desc; in Compute() local
370 output_desc.set_count(batch_size) in Compute()
398 ->ThenPoolForward(pooling_desc, input_desc, input_data, output_desc, in Compute()
404 output_desc, &output_data) in Compute()
Dcudnn_rnn_ops.cc696 std::unique_ptr<RnnSequenceTensorDescriptor>* output_desc, in CreateForwardAndBackwardIODescriptors() argument
766 *output_desc = output_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
772 *output_desc = output_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
779 *output_desc = output_desc_s.ConsumeValueOrDie(); in CreateForwardAndBackwardIODescriptors()
802 std::unique_ptr<RnnSequenceTensorDescriptor> output_desc; in DoForward() local
811 &output_desc, seq_lengths, time_major)); in DoForward()
833 params_data, *output_desc, &output_data, in DoForward()
865 std::unique_ptr<RnnSequenceTensorDescriptor> output_desc; in DoBackward() local
874 &output_desc, seq_lengths, time_major)); in DoBackward()
912 *c_state_desc, input_c_data, params_data, *output_desc, in DoBackward()
Dconv_ops_fused_impl.h597 se::dnn::BatchDescriptor output_desc;
598 output_desc.set_count(out_batch)
681 output_desc, &output_ptr_to_use, // output
693 output_desc, conv_desc, 1.0, 0.0, dnn_activation_mode,
/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()
179 &output_desc, in Curl_SOCKS5_gssapi_negotiate()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h82 const dnn::RnnSequenceTensorDescriptor& output_desc,
100 const dnn::RnnSequenceTensorDescriptor& output_desc,
118 const dnn::RnnSequenceTensorDescriptor& output_desc,
136 const dnn::RnnSequenceTensorDescriptor& output_desc,
161 const dnn::RnnSequenceTensorDescriptor& output_desc,
186 const dnn::RnnSequenceTensorDescriptor& output_desc,
577 const dnn::BatchDescriptor& output_desc,
648 const CudnnRnnSequenceTensorDescriptor& output_desc,
667 const CudnnRnnSequenceTensorDescriptor& output_desc,
Dcuda_dnn.cc1605 const CudnnRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward() argument
1636 if (!(output_desc.max_seq_length() == model_dims.max_seq_length && in ExtractAndCheckRnnForward()
1637 output_desc.batch_size() == model_dims.batch_size && in ExtractAndCheckRnnForward()
1638 output_desc.data_size() == in ExtractAndCheckRnnForward()
1754 const CudnnRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl() argument
1767 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
1813 /*yDesc=*/output_desc.data_handle(), in DoRnnForwardImpl()
1829 /*w=*/params.opaque(), /*yDesc=*/output_desc.handles(), in DoRnnForwardImpl()
1844 /*yDesc=*/output_desc.data_handle(), in DoRnnForwardImpl()
1862 /*w=*/params.opaque(), /*yDesc=*/output_desc.handles(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h109 const dnn::RnnSequenceTensorDescriptor& output_desc,
127 const dnn::RnnSequenceTensorDescriptor& output_desc,
145 const dnn::RnnSequenceTensorDescriptor& output_desc,
163 const dnn::RnnSequenceTensorDescriptor& output_desc,
188 const dnn::RnnSequenceTensorDescriptor& output_desc,
213 const dnn::RnnSequenceTensorDescriptor& output_desc,
581 const dnn::BatchDescriptor& output_desc,
746 const MIOpenRnnSequenceTensorDescriptor& output_desc,
763 const MIOpenRnnSequenceTensorDescriptor& output_desc,
Drocm_dnn.cc1980 const MIOpenRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward() argument
2009 if (!(output_desc.seq_length() == model_dims->seq_length && in ExtractAndCheckRnnForward()
2010 output_desc.batch_size() == model_dims->batch_size && in ExtractAndCheckRnnForward()
2011 output_desc.data_size() == in ExtractAndCheckRnnForward()
2090 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl() argument
2102 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2162 params.opaque() /*w*/, output_desc.handles() /*yDesc*/, in DoRnnForwardImpl()
2180 params.opaque() /*w*/, output_desc.handles() /*yDesc*/, in DoRnnForwardImpl()
2205 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnBackwardImpl() argument
2224 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2193 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward() argument
2214 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward() argument
2235 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward() argument
2297 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward() argument
2325 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward() argument
2353 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward() argument
2454 const dnn::BatchDescriptor& output_desc, in DoTransformTensor() argument
Dstream.h1787 const dnn::RnnSequenceTensorDescriptor &output_desc,
1806 const dnn::RnnSequenceTensorDescriptor &output_desc,
1824 const dnn::RnnSequenceTensorDescriptor &output_desc,
1845 const dnn::RnnSequenceTensorDescriptor &output_desc,
1870 const dnn::RnnSequenceTensorDescriptor &output_desc,
1895 const dnn::RnnSequenceTensorDescriptor &output_desc,
1929 const dnn::BatchDescriptor &output_desc,
1938 const dnn::BatchDescriptor &output_desc, in ThenTransformTensor() argument
1941 input_data, output_desc, in ThenTransformTensor()
Dstream.cc4547 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward() argument
4560 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4580 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward() argument
4593 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4614 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward() argument
4627 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4648 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward() argument
4668 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
4692 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward() argument
4712 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
[all …]

12