Home
last modified time | relevance | path

Searched defs:output_desc (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbroadcast_to.h25 const NdArrayDesc<N>& output_desc, char* output_data, in BroadcastImpl()
67 NdArrayDesc<N> output_desc; in BroadcastTo() local
Dsub.h82 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
122 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
179 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
216 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
272 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
309 NdArrayDesc<N> output_desc; in BroadcastSubSlow() local
346 NdArrayDesc<N> output_desc; in BroadcastSub16POTSlow() local
Dmaximum_minimum.h46 NdArrayDesc<N> output_desc; in MaximumMinimumBroadcastSlow() local
Ddiv.h87 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local
149 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local
Dtranspose.h39 NdArrayDesc<N> output_desc; in TransposeImpl() local
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc133 const se::dnn::BatchDescriptor& output_desc, in LogConvAutotuneResults()
176 const se::dnn::BatchDescriptor& output_desc, in LogFusedConvForwardAutotuneResults()
Dcudnn_pooling_gpu.cc77 se::dnn::BatchDescriptor output_desc(3); in Compute() local
Dconv_grad_input_ops.cc255 se::dnn::BatchDescriptor output_desc; in operator ()() local
Dconv_ops_3d.cc404 se::dnn::BatchDescriptor output_desc(3); in launch() local
Dpooling_ops_common.cc369 se::dnn::BatchDescriptor output_desc; in Compute() local
Dcudnn_rnn_ops.cc696 std::unique_ptr<RnnSequenceTensorDescriptor>* output_desc, in CreateForwardAndBackwardIODescriptors()
802 std::unique_ptr<RnnSequenceTensorDescriptor> output_desc; in DoForward() local
865 std::unique_ptr<RnnSequenceTensorDescriptor> output_desc; in DoBackward() local
Dconv_grad_ops_3d.cc1305 se::dnn::BatchDescriptor output_desc(3); in Compute() local
1802 se::dnn::BatchDescriptor output_desc(3); in Compute() local
Dconv_grad_filter_ops.cc860 se::dnn::BatchDescriptor output_desc; in operator ()() local
Dconv_ops.cc885 se::dnn::BatchDescriptor output_desc; in operator ()() local
/external/curl/lib/
Dsocks_sspi.c77 SecBufferDesc input_desc, output_desc, wrap_desc; in Curl_SOCKS5_gssapi_negotiate() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2193 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2214 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2235 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2297 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2325 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2353 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2454 const dnn::BatchDescriptor& output_desc, in DoTransformTensor()
Dstream.h1938 const dnn::BatchDescriptor &output_desc, in ThenTransformTensor()
Dstream.cc4547 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward()
4580 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward()
4614 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnForward()
4648 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward()
4692 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward()
4737 const dnn::RnnSequenceTensorDescriptor &output_desc, in ThenRnnBackward()
4809 const dnn::BatchDescriptor &output_desc, in ThenTransformTensor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1605 const CudnnRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward()
1754 const CudnnRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl()
1894 const CudnnRnnSequenceTensorDescriptor& output_desc, in DoRnnBackwardImpl()
2135 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2176 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2218 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2260 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2311 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2363 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
3985 const dnn::BatchDescriptor& output_desc, in DoTransformTensor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1980 const MIOpenRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward()
2090 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl()
2205 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnBackwardImpl()
2586 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2628 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2671 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnForward()
2693 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2744 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
2796 const dnn::RnnSequenceTensorDescriptor& output_desc, in DoRnnBackward()
3744 const dnn::BatchDescriptor& output_desc, in DoTransformTensor()
/external/mesa3d/src/gallium/auxiliary/translate/
Dtranslate_sse.c465 const struct util_format_description *output_desc = in translate_attr_convert() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2637 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local
2681 NdArrayDesc<N> output_desc; in BroadcastDivSlow() local