| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_dynamic_shapes.cpp | 54 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
140 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
231 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
313 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
370 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
497 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
559 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
656 std::vector<torch::jit::StrideInput> input_desc = { in TEST() local
|
| H A D | test_kernel.cpp | 1928 std::vector<torch::jit::StrideInput> input_desc = { in TEST_F() local
2003 std::vector<torch::jit::StrideInput> input_desc = { in TEST_F() local
2080 std::vector<torch::jit::StrideInput> input_desc = { in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A 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
|
| H A D | transpose.h | 38 NdArrayDesc<N> input_desc; in TransposeImpl() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/cuda/ |
| H A D | fused_kernel.cpp | 88 std::vector<TensorDesc> input_desc, in FusedKernelCUDA()
249 std::vector<TensorDesc> input_desc, in createFusionKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | gpu_utils.cc | 142 const se::dnn::BatchDescriptor& input_desc, in LogConvAutotuneResults()
185 const se::dnn::BatchDescriptor& input_desc, in LogFusedConvForwardAutotuneResults()
|
| H A D | conv_ops_gpu.cc | 110 const se::dnn::BatchDescriptor& input_desc, in AutotuneFusedConv()
276 se::dnn::ConvolutionKind kind, const se::dnn::BatchDescriptor& input_desc, in AutotuneUnfusedConv()
|
| H A D | conv_ops_gpu.h | 164 const se::dnn::BatchDescriptor& input_desc, in LaunchAutotunedConv()
|
| H A D | cudnn_pooling_gpu.cc | 74 se::dnn::BatchDescriptor input_desc(3); in Compute() local
|
| H A D | conv_grad_input_ops.cc | 245 se::dnn::BatchDescriptor input_desc; in operator ()() local
|
| H A D | cudnn_rnn_ops.cc | 696 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
|
| H A D | conv_ops_3d.cc | 405 se::dnn::BatchDescriptor input_desc(3); in launch() local
|
| H A D | conv_grad_ops_3d.cc | 1400 se::dnn::BatchDescriptor input_desc(3); in Compute() local
1796 se::dnn::BatchDescriptor input_desc(3); in Compute() local
|
| H A D | pooling_ops_common.cc | 365 se::dnn::BatchDescriptor input_desc; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 2114 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()
|
| H A D | stream.cc | 2032 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()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/cpu/ |
| H A D | fused_kernel.cpp | 309 std::vector<TensorDesc> input_desc, in FusedKernelCPU()
344 std::vector<TensorDesc> input_desc, in createFusionKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 2028 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 …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1464 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 …]
|
| /aosp_15_r20/external/curl/lib/ |
| H A D | socks_sspi.c | 80 SecBufferDesc input_desc, output_desc, wrap_desc; in Curl_SOCKS5_gssapi_negotiate() local
|
| /aosp_15_r20/external/curl/lib/vauth/ |
| H A D | krb5_sspi.c | 265 SecBufferDesc input_desc; in Curl_auth_create_gssapi_security_message() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/ |
| H A D | compiler.cpp | 205 const std::vector<TensorDesc>& input_desc = arg_spec.descs(); in compileKernel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | rebatch_dataset_op.cc | 166 for (const auto& input_desc : input_descriptors_) { in GetNextInternal() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | symbolic_shape_runtime_fusion.cpp | 566 std::vector<StrideInput> input_desc; in __anonc321d2750602() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear_dynamic.cpp | 514 auto input_desc = ideep::tensor::desc(input_dims, input_data_type); in apply_dynamic_impl() local
|