| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | broadcast_to.h | 24 void BroadcastImpl(const NdArrayDesc<N>& input_desc, const char* input_data, in BroadcastImpl()
66 NdArrayDesc<N> input_desc; in BroadcastTo() local
|
| D | transpose.h | 38 NdArrayDesc<N> input_desc; in TransposeImpl() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 131 const se::dnn::BatchDescriptor& input_desc, in LogConvAutotuneResults()
174 const se::dnn::BatchDescriptor& input_desc, in LogFusedConvForwardAutotuneResults()
|
| D | cudnn_pooling_gpu.cc | 73 se::dnn::BatchDescriptor input_desc(3); in Compute() local
|
| D | conv_grad_input_ops.cc | 249 se::dnn::BatchDescriptor input_desc; in operator ()() local
|
| D | conv_ops_3d.cc | 397 se::dnn::BatchDescriptor input_desc(3); in launch() local
|
| D | pooling_ops_common.cc | 362 se::dnn::BatchDescriptor input_desc; in Compute() local
|
| D | cudnn_rnn_ops.cc | 693 std::unique_ptr<RnnSequenceTensorDescriptor>* input_desc, in CreateForwardAndBackwardIODescriptors()
799 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoForward() local
862 std::unique_ptr<RnnSequenceTensorDescriptor> input_desc; in DoBackward() local
|
| D | conv_grad_ops_3d.cc | 1298 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1792 se::dnn::BatchDescriptor input_desc(3); in Compute() local
|
| D | conv_grad_filter_ops.cc | 854 se::dnn::BatchDescriptor input_desc; in operator ()() local
|
| D | conv_ops.cc | 879 se::dnn::BatchDescriptor input_desc; in operator ()() local
|
| /external/curl/lib/ |
| D | socks_sspi.c | 77 SecBufferDesc input_desc, output_desc, wrap_desc; in Curl_SOCKS5_gssapi_negotiate() local
|
| /external/curl/lib/vauth/ |
| D | krb5_sspi.c | 289 SecBufferDesc input_desc; in Curl_auth_create_gssapi_security_message() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1286 const TensorDescriptor& input_desc, const FilterDescriptor& filter_desc, in CheckAndFetchProjectionWeights()
1360 TensorDescriptor input_desc = CreateTensorDescriptor(); in Create() local
1599 const CudnnRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward()
1660 const CudnnRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize()
1677 const CudnnRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace()
1748 const CudnnRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl()
1888 const CudnnRnnSequenceTensorDescriptor& input_desc, in DoRnnBackwardImpl()
2128 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2170 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2211 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_op.h | 112 for (const auto& input_desc : input_descs) { in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2186 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2207 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2228 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2290 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
2318 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
2346 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
2451 const dnn::BatchDescriptor& input_desc, in DoTransformTensor()
|
| D | stream.h | 1936 Stream &ThenTransformTensor(const dnn::BatchDescriptor &input_desc, in ThenTransformTensor()
|
| D | stream.cc | 4540 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward()
4574 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward()
4607 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnForward()
4641 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward()
4686 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward()
4730 const dnn::RnnSequenceTensorDescriptor &input_desc, in ThenRnnBackward()
4806 Stream &Stream::ThenTransformTensor(const dnn::BatchDescriptor &input_desc, in ThenTransformTensor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1974 const MIOpenRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward()
2034 const MIOpenRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize()
2050 const MIOpenRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace()
2084 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl()
2199 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnBackwardImpl()
2318 miopenTensorDescriptor_t input_desc = nullptr; in MIOpenRnnParamsDescriptor() local
2579 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2622 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2664 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnForward()
2686 const dnn::RnnSequenceTensorDescriptor& input_desc, in DoRnnBackward()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.cc | 66 se::dnn::BatchDescriptor input_desc; member
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | rebatch_dataset_op.cc | 160 for (const auto& input_desc : input_descriptors_) { in GetNextInternal() local
|
| /external/mesa3d/src/gallium/auxiliary/translate/ |
| D | translate_sse.c | 463 const struct util_format_description *input_desc = in translate_attr_convert() local
|