Home
last modified time | relevance | path

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

12

/external/XNNPACK/src/subgraph/
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
Dsigmoid.c209 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_sigmoid() local
Ddepth-to-space.c220 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_depth_to_space() local
Delu.c189 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_elu() local
Dstatic-reshape.c197 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_reshape() local
Dleaky-relu.c211 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_leaky_relu() local
Dstatic-resize-bilinear-2d.c275 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_resize_bilinear_2d() local
Dstatic-transpose.c204 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_transpose() local
Dstatic-constant-pad.c202 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_constant_pad() local
Dclamp.c218 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_clamp() local
Dadd2.c292 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_add2() local
Dsubtract.c292 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_subtract() local
Dmultiply2.c296 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_multiply2() local
Dmax-pooling-2d.c338 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_max_pooling_2d() local
Dglobal-average-pooling.c268 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in define_global_average_pooling_nd() local
Dfully-connected.c454 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_fully_connected() local
Ddeconvolution-2d.c576 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_deconvolution_2d() local
Dconvolution-2d.c688 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_convolution_2d() local
Ddepthwise-convolution-2d.c681 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_depthwise_convolution_2d() local
Deven-split.c544 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_even_split_n() local
Dconcatenate.c562 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_concatenate_n() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
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
Dcuda_dnn.cc1239 cudnnDataType_t compute_type, in CudnnRnnDescriptor()
1267 cudnnDataType_t data_type, cudnnDataType_t compute_type, in Create()
1389 cudnnDataType_t compute_type() const { return compute_type_; } in compute_type() function in stream_executor::gpu::CudnnRnnDescriptor
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc162 auto compute_type = data_type; // use the data_type for compute as well. in CreateTfrtOps() local
/external/XNNPACK/src/xnnpack/
Dsubgraph.h177 enum xnn_compute_type compute_type; member

12