| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ops_nccl.py | 418 def allgather_base(output_t, input_t): argument
440 def allgather_base(output_t, input_t): argument
474 def gather(output_t, input_t, rootRank): argument
509 def gather(output_t, input_t, rootRank): argument
587 def scatter(output_t, input_t, rootRank): argument
622 def scatter(output_t, input_t, rootRank): argument
705 def reduce_scatter_base(output_t, input_t): argument
857 def reduce_scatter_base(output_t, input_t): argument
961 def allgather_base(output_t, input_t): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | ConvShared.cpp | 253 Tensor output_t = at::detail::empty_cuda( in cudnn_convolution() local
286 Tensor& output_t) { in cudnn_convolution_out()
323 Tensor output_t = at::detail::empty_cuda( in cudnn_convolution_transpose_backward_input() local
481 auto output_t = cudnn_convolution_transpose_forward( in cudnn_convolution_transpose() local
718 Tensor output_t = at::detail::empty_cuda( in cudnn_convolution_relu() local
777 Tensor output_t = at::detail::empty_cuda( in cudnn_convolution_add_relu() local
|
| H A D | GridSampler.cpp | 97 auto output_t = at::empty({0}, input->options()); in cudnn_grid_sampler_forward() local
|
| H A D | BatchNorm.cpp | 171 auto output_t = in cudnn_batch_norm() local
|
| H A D | ConvPlaceholders.cpp | 51 Tensor& output_t) { in cudnn_convolution_out()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | searchsorted_op.cc | 118 Tensor* output_t; in Compute() local
171 Tensor* output_t; in Compute() local
|
| H A D | matching_files_op.cc | 52 Tensor* output_t = nullptr; in Compute() local
|
| H A D | population_count_op.cc | 47 Tensor* output_t; in Compute() local
|
| H A D | one_hot_op.cc | 114 auto output_t = in Compute() local
|
| H A D | edit_distance_op.cc | 176 auto output_t = output->flat<float>(); in Compute() local
|
| H A D | shape_ops.h | 178 Tensor output_t; in Compute() local
|
| H A D | slice_op.cc | 174 auto output_t = result->tensor<T, 2>(); in Compute() local
|
| H A D | list_kernels.cc | 552 auto output_t = output->flat<Variant>(); in Compute() local
|
| H A D | bincount_op.cc | 224 Tensor* output_t; in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | CopyKernel.cpp | 18 static bool reduced_input(ScalarType input_t, ScalarType output_t) { in reduced_input()
23 static bool reduced_output(ScalarType input_t, ScalarType output_t) { in reduced_output()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | Conv_miopen.cpp | 768 Tensor output_t = at::detail::empty_cuda( in miopen_convolution_forward() local
810 auto output_t = miopen_convolution_forward( in miopen_convolution() local
876 Tensor output_t = at::detail::empty_cuda( in miopen_depthwise_convolution_forward() local
911 auto output_t = miopen_depthwise_convolution_forward( in miopen_depthwise_convolution() local
1478 auto output_t = miopen_convolution_transpose_forward( in miopen_convolution_transpose() local
1627 Tensor output_t = at::detail::empty_cuda( in miopen_convolution_relu() local
|
| H A D | BatchNorm_miopen.cpp | 103 auto output_t = at::empty(input->sizes(), input->options()); in miopen_batch_norm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | softmax_op.cc | 81 Tensor output_t(cpu_allocator(), DT_VARIANT, TensorShape({})); in Compute() local
|
| H A D | transpose_op.cc | 115 Tensor output_t(cpu_allocator(), DT_VARIANT, TensorShape({})); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | tf_op_shim.cc | 94 tensorflow::Tensor* output_t = nullptr; in GetOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | image_ops.cc | 98 Tensor* output_t; in DoImageProjectiveTransformOp() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Loops.cuh | 276 using output_t = typename traits::result_type; in gpu_kernel_multiple_outputs_impl() typedef
|
| H A D | DistributionTemplates.h | 239 using output_t = typename function_traits<func_t>::result_type; in distribution_binary_kernel() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_shape_ops.py | 406 def test_flip_impl(input_t, dims, output_t): argument
|
| /aosp_15_r20/external/mesa3d/src/util/blake3/ |
| H A D | blake3.c | 60 } output_t; typedef
|