Home
last modified time | relevance | path

Searched refs:getTensorInfo (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cu771 getTensorInfo<const scalar_t, int>(input), in launch_grid_sampler_2d_forward_kernel()
772 getTensorInfo<const scalar_t, int>(grid), in launch_grid_sampler_2d_forward_kernel()
773 getTensorInfo<scalar_t, int>(output), in launch_grid_sampler_2d_forward_kernel()
782 getTensorInfo<const scalar_t, int64_t>(input), in launch_grid_sampler_2d_forward_kernel()
783 getTensorInfo<const scalar_t, int64_t>(grid), in launch_grid_sampler_2d_forward_kernel()
784 getTensorInfo<scalar_t, int64_t>(output), in launch_grid_sampler_2d_forward_kernel()
816 getTensorInfo<const scalar_t, int>(input), in launch_grid_sampler_3d_forward_kernel()
817 getTensorInfo<const scalar_t, int>(grid), in launch_grid_sampler_3d_forward_kernel()
818 getTensorInfo<scalar_t, int>(output), in launch_grid_sampler_3d_forward_kernel()
827 getTensorInfo<const scalar_t, int64_t>(input), in launch_grid_sampler_3d_forward_kernel()
[all …]
H A DRNN.cu27 using at::cuda::detail::getTensorInfo;
67 return t.defined() ? getTensorInfo<T, T2>(t) : TensorInfo<T, T2>{}; in tryGetTensorInfo()
381 auto input_gatesI = getTensorInfo<scalar_t, index_type>(input_gates); in lstm_forward_impl()
382 auto hidden_gatesI = getTensorInfo<scalar_t, index_type>(hidden_gates); in lstm_forward_impl()
385 auto cxI = getTensorInfo<scalar_t, index_type>(cx); in lstm_forward_impl()
386 auto hyI = getTensorInfo<scalar_t, index_type>(hy); in lstm_forward_impl()
387 auto cyI = getTensorInfo<scalar_t, index_type>(cy); in lstm_forward_impl()
388 auto workspaceI = getTensorInfo<scalar_t, index_type>(workspace); in lstm_forward_impl()
420 auto cxI = getTensorInfo<scalar_t, index_type>(cx); in lstm_backward_impl()
421 auto cyI = getTensorInfo<scalar_t, index_type>(cy); in lstm_backward_impl()
[all …]
H A DIndexing.cu923 cuda::detail::getTensorInfo<scalar_t, unsigned int>(self_); in index_add_cuda_impl()
929 cuda::detail::getTensorInfo<const scalar_t, unsigned int>(source_); in index_add_cuda_impl()
934 cuda::detail::getTensorInfo<const index_t, unsigned int>(index); in index_add_cuda_impl()
975 cuda::detail::getTensorInfo<scalar_t, uint64_t>(self_); in index_add_cuda_impl()
981 cuda::detail::getTensorInfo<const scalar_t, uint64_t>(source_); in index_add_cuda_impl()
987 cuda::detail::getTensorInfo<const index_t, uint64_t>(index); in index_add_cuda_impl()
1096 cuda::detail::getTensorInfo<scalar_t, unsigned int>(self_); in index_reduce_func_cuda_impl()
1102 cuda::detail::getTensorInfo<const scalar_t, unsigned int>(source_); in index_reduce_func_cuda_impl()
1107 cuda::detail::getTensorInfo<const index_t, unsigned int>(index); in index_reduce_func_cuda_impl()
1148 cuda::detail::getTensorInfo<scalar_t, uint64_t>(self_); in index_reduce_func_cuda_impl()
[all …]
H A DTriangularOps.cu126 auto result_info = cuda::detail::getTensorInfo<scalar_t, int32_t>(result); in triu_tril_cuda_template()
127 auto self_info = cuda::detail::getTensorInfo<const scalar_t, int32_t>(self); in triu_tril_cuda_template()
135 auto result_info = cuda::detail::getTensorInfo<scalar_t, int64_t>(result); in triu_tril_cuda_template()
136 auto self_info = cuda::detail::getTensorInfo<const scalar_t, int64_t>(self); in triu_tril_cuda_template()
H A DSort.cu313 at::cuda::detail::getTensorInfo<scalar_t, unsigned int>(key); in sortCommon()
315 at::cuda::detail::getTensorInfo<int64_t, unsigned int>(value); in sortCommon()
341 at::cuda::detail::getTensorInfo<scalar_t, uint64_t>(key); in sortCommon()
343 at::cuda::detail::getTensorInfo<int64_t, uint64_t>(value); in sortCommon()
H A DSortingCommon.cuh119 auto self_info = cuda::detail::getTensorInfo<const scalar_t, index_t>(self); in run_launcher()
120 auto values_info = cuda::detail::getTensorInfo<scalar_t, index_t>(values); in run_launcher()
121 auto indices_info = cuda::detail::getTensorInfo<int64_t, index_t>(indices); in run_launcher()
H A DSummaryOps.cu221 auto aInfo = detail::getTensorInfo<output_t, IndexType>(a); in CUDA_tensor_histogram()
222 auto bInfo = detail::getTensorInfo<const input_t, IndexType>(b); in CUDA_tensor_histogram()
226 auto cInfo = detail::getTensorInfo<output_t, IndexType>(c); in CUDA_tensor_histogram()
H A DDropout.cu239 cuda::detail::getTensorInfo<const scalar_t, index_type>(self); in launcher()
241 cuda::detail::getTensorInfo<scalar_t, index_type>(ret); in launcher()
243 cuda::detail::getTensorInfo<mask_t, index_type>(mask); in launcher()
H A DTensorModeKernel.cu227 auto ti_values = cuda::detail::getTensorInfo<scalar_t, unsigned int>(values); in fused_mode()
228 auto ti_indices = cuda::detail::getTensorInfo<int64_t, unsigned int>(indices); in fused_mode()
H A DTensorTopK.cu840 at::cuda::detail::getTensorInfo<const scalar_t, INDEX_T>(input); \ in launch_gather_topk_kernel()
842 at::cuda::detail::getTensorInfo<scalar_t, INDEX_T>(values); \ in launch_gather_topk_kernel()
844 at::cuda::detail::getTensorInfo<int64_t, INDEX_T>(indices); \ in launch_gather_topk_kernel()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dconv2d.cpp370 auto inpInfo = getTensorInfo(inp); in computeConv2d()
371 auto wInfo = getTensorInfo(w); in computeConv2d()
372 auto bInfo = getTensorInfo(b); in computeConv2d()
418 auto inpInfo = getTensorInfo(inp); in computeConv1d()
419 auto wInfo = getTensorInfo(w); in computeConv1d()
420 auto bInfo = getTensorInfo(b); in computeConv1d()
H A Dmisc.h13 std::optional<TensorInfo> getTensorInfo(const BufHandle& b);
H A Dmisc.cpp159 std::optional<TensorInfo> getTensorInfo(const BufHandle& b) { in getTensorInfo() function
335 auto buf_info = getTensorInfo(b); in computeChunk()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDATensorMath.cu51 #define I_INFO(tensor) cuda::detail::getTensorInfo<int64_t, uint64_t>(tensor)
52 #define V_INFO(tensor) cuda::detail::getTensorInfo<scalar_t, uint64_t>(tensor)
58 using at::cuda::detail::getTensorInfo;
646 auto grad_indices_ti = getTensorInfo<int64_t, int64_t>(grad_indices_1D); in _sparse_sum_backward_cuda()
647 auto input_indices_ti = getTensorInfo<int64_t, int64_t>(input_indices_1D); in _sparse_sum_backward_cuda()
648 auto input_indices_pos_ti = getTensorInfo<int64_t, int64_t>(input_indices_pos); in _sparse_sum_backward_cuda()
651 auto grad_values_expand_ti = getTensorInfo<scalar_t, int64_t>(grad_values_expand); in _sparse_sum_backward_cuda()
652 auto grad_input_values_ti = getTensorInfo<scalar_t, int64_t>(grad_input_values); in _sparse_sum_backward_cuda()
722 auto indices_1D_ti = getTensorInfo<int64_t, int64_t>(indices_1D); in search_end_matrix_indices()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAApplyUtils.cuh480 detail::getTensorInfo<scalar1, unsigned int>(a); in CUDA_tensor_apply2()
483 detail::getTensorInfo<scalar2, unsigned int>(b); in CUDA_tensor_apply2()
491 detail::getTensorInfo<scalar1, uint64_t>(a); in CUDA_tensor_apply2()
494 detail::getTensorInfo<scalar2, uint64_t>(b); in CUDA_tensor_apply2()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactory.cpp812 auto getTensorInfo = [&dataType](const InputSlot& slot) in IsLayerConfigurationSupported() local
817 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
818 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
1186 auto getTensorInfo = [&dataType](const OutputSlot& slot) in IsLayerConfigurationSupported() local
1190 auto beginI = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
1191 auto endI = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
1207 auto getTensorInfo = [&dataType](const InputSlot& slot) in IsLayerConfigurationSupported() local
1211 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
1212 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DIndexUtils.cuh14 getTensorInfo(const at::TensorBase &t) { in getTensorInfo() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/turbine/
Dtensorflow_core_proto_java_lite.jarorg/tensorflow/distruntime/ClusterDef$Builder.class <Unknown> package org.tensorflow. ...
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/javac/
Dtensorflow_core_proto_java_lite.jarMETA-INF/ META-INF/MANIFEST.MF com/ com/google/ com/ ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/combined/
DFederatedCompute.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx/ META- ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/federatedcompute/apk/FederatedCompute/android_common_apex33/jarjar/
DFederatedCompute.jarxla/Protocol.class Protocol.java package xla public final xla.Protocol extends java.lang ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/OnDevicePersonalization/OnDevicePersonalization/android_common_apex33/withres/
DOnDevicePersonalization.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx.annotation_annotation- ...