Home
last modified time | relevance | path

Searched defs:compute_type (Results 1 – 25 of 52) sorted by relevance

123

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/
H A DGemmRocblas.h143 auto compute_type = RocBlasComputeTypeFor<T>(); in Call() local
176 auto compute_type = RocBlasComputeTypeFor<T>(); in GetRocBlasGemmTypeStringAndOps() local
212 auto compute_type = RocBlasComputeTypeFor<T>(); in Call() local
246 auto compute_type = RocBlasComputeTypeFor<T>(); in GetRocBlasGemmStridedBatchedTypeStringAndOps() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_div.cpp55 ScalarType compute_type = utils::get_compute_type(common_type); in div_out() local
116 ScalarType compute_type = utils::get_compute_type(common_type); in div_out_mode() local
184 ScalarType compute_type = utils::get_compute_type(common_type); in div_scalar_out() local
248 ScalarType compute_type = utils::get_compute_type(common_type); in div_scalar_mode_out() local
H A Dop_pow.cpp47 ScalarType compute_type = utils::get_compute_type(common_type); in pow_Tensor_Tensor_out() local
97 ScalarType compute_type = utils::get_compute_type(common_type); in pow_Tensor_Scalar_out() local
144 ScalarType compute_type = utils::get_compute_type(common_type); in pow_Scalar_out() local
H A Dop_mul.cpp42 ScalarType compute_type = utils::get_compute_type(common_type); in mul_out() local
91 ScalarType compute_type = utils::get_compute_type(common_type); in mul_scalar_out() local
H A Dop_remainder.cpp48 ScalarType compute_type = utils::get_compute_type(common_type); in remainder_Tensor_out() local
122 ScalarType compute_type = utils::get_compute_type(common_type); in remainder_Scalar_out() local
H A Dop_fmod.cpp47 ScalarType compute_type = utils::get_compute_type(common_type); in fmod_Tensor_out() local
124 ScalarType compute_type = utils::get_compute_type(common_type); in fmod_Scalar_out() local
H A Dop_add.cpp48 ScalarType compute_type = utils::get_compute_type(common_type); in add_out() local
97 ScalarType compute_type = utils::get_compute_type(common_type); in add_scalar_out() local
H A Dop_sub.cpp52 ScalarType compute_type = utils::get_compute_type(common_type); in sub_out() local
105 ScalarType compute_type = utils::get_compute_type(common_type); in sub_scalar_out() local
H A Dop_clamp.cpp131 ScalarType compute_type = utils::get_compute_type(common_type); in clamp_out() local
207 ScalarType compute_type = utils::get_compute_type(common_type); in clamp_tensor_out() local
H A Dop_sigmoid.cpp39 ScalarType compute_type = in sigmoid_out() local
H A Dop_minimum.cpp42 ScalarType compute_type = utils::get_compute_type(common_type); in minimum_out() local
H A Dop_maximum.cpp42 ScalarType compute_type = utils::get_compute_type(common_type); in maximum_out() local
H A Dop_floor_divide.cpp48 ScalarType compute_type = utils::get_compute_type(common_type); in floor_divide_out() local
H A Dop_where.cpp40 ScalarType compute_type = utils::get_compute_type(common_type); in where_out() local
H A Dop_rsub.cpp47 ScalarType compute_type = utils::get_compute_type(common_type); in rsub_scalar_out() local
H A Dop_atan2.cpp52 ScalarType compute_type = utils::get_compute_type(common_type); in atan2_out() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Delementwise_util.h55 constexpr auto compute_type = CppTypeToScalarType<CTYPE_COMMON>::value; in apply_unitensor_elementwise_fn() local
95 constexpr auto compute_type = CppTypeToScalarType<CTYPE_COMMON>::value; in apply_bitensor_elementwise_fn() local
178 constexpr auto compute_type = CppTypeToScalarType<CTYPE_COMMON>::value; in apply_tritensor_elementwise_fn() local
242 ScalarType compute_type = common_type; in get_compute_type() local
H A Ddtype_util.cpp20 const ScalarType compute_type) { in check_tensor_dtype()
/aosp_15_r20/external/executorch/kernels/portable/cpu/pattern/
H A Dbitwise_op.h79 ScalarType compute_type = utils::get_compute_type(common_type); in bitwise_tensor_out() local
119 ScalarType compute_type = utils::get_compute_type(common_type); in bitwise_scalar_out() local
H A Dcomparison_op.h91 ScalarType compute_type = utils::get_compute_type(common_type); in comparison_tensor_out() local
126 ScalarType compute_type = utils::get_compute_type(common_type); in comparison_scalar_out() local
/aosp_15_r20/external/XNNPACK/src/subgraph/
H A Dconvert.c253 enum xnn_compute_type compute_type, in xnn_init_convert_node()
331 …enum xnn_compute_type compute_type = validate_datatypes(input_value->datatype, output_value->datat… in xnn_define_convert() local
H A Dsigmoid.c209 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_sigmoid() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas_lt.cc177 blas::ComputationType compute_type, blas::DataType scale_type, in Create()
196 cublasComputeType_t BlasLt::MatmulDesc::compute_type() const { in compute_type() function in stream_executor::cuda::BlasLt::MatmulDesc
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp271 cublasComputeType_t compute_type, in CuBlasLtMatmulDescriptor()
578 auto compute_type = CUBLAS_COMPUTE_32F; in bgemm_internal_cublas() local
580 auto compute_type = CUDA_R_32F; in bgemm_internal_cublas() local
964 auto compute_type = CUBLAS_COMPUTE_32F; in gemm_internal_cublas() local
966 auto compute_type = CUDA_R_32F; in gemm_internal_cublas() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseBlasImpl.cpp732 auto compute_type = at::cuda::getCudaDataType<scalar_t>(); in spgemm() local
1022 cudaDataType compute_type = at::cuda::getCudaDataType<scalar_t>(); in addmv_out_sparse_csr() local
1300 cudaDataType compute_type = at::cuda::getCudaDataType<scalar_t>(); in triangular_solve_out_sparse_csr() local

123