Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbroadcast_to.h24 void BroadcastImpl(const NdArrayDesc<N>& input_desc, const char* input_data, in BroadcastImpl()
66 NdArrayDesc<N> input_desc; in BroadcastTo() local
Dtranspose.h38 NdArrayDesc<N> input_desc; in TransposeImpl() local
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc142 const se::dnn::BatchDescriptor& input_desc, in LogConvAutotuneResults()
185 const se::dnn::BatchDescriptor& input_desc, in LogFusedConvForwardAutotuneResults()
Dconv_ops_gpu.cc110 const se::dnn::BatchDescriptor& input_desc, in AutotuneFusedConv()
276 se::dnn::ConvolutionKind kind, const se::dnn::BatchDescriptor& input_desc, in AutotuneUnfusedConv()
Dconv_ops_gpu.h164 const se::dnn::BatchDescriptor& input_desc, in LaunchAutotunedConv()
Dcudnn_pooling_gpu.cc74 se::dnn::BatchDescriptor input_desc(3); in Compute() local
Dconv_grad_input_ops.cc245 se::dnn::BatchDescriptor input_desc; in operator ()() local
Dconv_ops_3d.cc405 se::dnn::BatchDescriptor input_desc(3); in launch() local
Dpooling_ops_common.cc365 se::dnn::BatchDescriptor input_desc; in Compute() local
Dconv_grad_ops_3d.cc1400 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1796 se::dnn::BatchDescriptor input_desc(3); in Compute() local
Dcudnn_rnn_ops.cc696 std::unique_ptr<RnnSequenceTensorDescriptor>* input_desc, in CreateForwardAndBackwardIODescriptors()
802 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoForward() local
885 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoBackward() local
Dconv_grad_filter_ops.cc864 se::dnn::BatchDescriptor input_desc; in operator ()() local
Dconv_ops.cc1030 se::dnn::BatchDescriptor input_desc; in operator ()() local
/external/curl/lib/
Dsocks_sspi.c80 SecBufferDesc input_desc, output_desc, wrap_desc; in Curl_SOCKS5_gssapi_negotiate() local
/external/curl/lib/vauth/
Dkrb5_sspi.c265 SecBufferDesc input_desc; in Curl_auth_create_gssapi_security_message() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h2114 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2136 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2158 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2221 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
2250 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
2279 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
2385 const dnn::BatchDescriptor& input_desc, in DoTransformTensor()
Dstream.cc2032 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward()
2068 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward()
2103 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward()
2139 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward()
2185 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward()
2230 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward()
2307 Stream &Stream::ThenTransformTensor(const dnn::BatchDescriptor &input_desc, in ThenTransformTensor()
Dstream.h1389 Stream &ThenTransformTensor(const dnn::BatchDescriptor &input_desc, in ThenTransformTensor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2028 const MIOpenRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward()
2088 const MIOpenRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize()
2104 const MIOpenRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace()
2138 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl()
2253 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnBackwardImpl()
2372 miopenTensorDescriptor_t input_desc = nullptr; in MIOpenRnnParamsDescriptor() local
2633 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2677 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2720 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2743 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1464 const TensorDescriptor& input_desc, const FilterDescriptor& filter_desc, in CheckAndFetchProjectionWeights()
1538 TensorDescriptor input_desc = CreateTensorDescriptor(); in Create() local
1774 const CudnnRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward()
1835 const CudnnRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize()
1858 const CudnnRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace()
1931 const CudnnRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl()
2144 const CudnnRnnSequenceTensorDescriptor& input_desc, in DoRnnBackwardImpl()
2471 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2515 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2558 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc166 for (const auto& input_desc : input_descriptors_) { in GetNextInternal() local
/external/mesa3d/src/gallium/auxiliary/translate/
Dtranslate_sse.c463 const struct util_format_description *input_desc = in translate_attr_convert() local