Home
last modified time | relevance | path

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

/external/XNNPACK/src/subgraph/
Dconvert.c205 enum xnn_compute_type compute_type, in xnn_init_convert_node()
292 …enum xnn_compute_type compute_type = validate_datatypes(input_value->datatype, output_value->datat… in xnn_define_convert() local
Dstatic-reshape.c190 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_reshape() local
Dsigmoid.c197 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_sigmoid() local
Delu.c177 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_elu() local
Dstatic-resize-bilinear-2d.c258 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_resize_bilinear_2d() local
Dstatic-constant-pad.c204 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_static_constant_pad() local
Dclamp.c207 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_clamp() local
Dglobal-average-pooling-2d.c281 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_global_average_pooling_2d() local
Dsubtract.c303 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_subtract() local
Dmultiply2.c303 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_multiply2() local
Dadd2.c320 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_add2() local
Dmax-pooling-2d.c348 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_max_pooling_2d() local
Dfully-connected.c450 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_fully_connected() local
Ddeconvolution-2d.c558 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_deconvolution_2d() local
Ddepthwise-convolution-2d.c701 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_depthwise_convolution_2d() local
Dconvolution-2d.c708 enum xnn_compute_type compute_type = xnn_compute_type_invalid; in xnn_define_convolution_2d() local
/external/ComputeLibrary/src/core/GLES_COMPUTE/kernels/
DGCPixelWiseMultiplicationKernel.cpp73 std::string compute_type; in configure() local
/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.h176 enum xnn_compute_type compute_type; member
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1166 cudnnDataType_t compute_type, in CudnnRnnDescriptor()
1194 cudnnDataType_t data_type, cudnnDataType_t compute_type, in Create()
1316 cudnnDataType_t compute_type() const { return compute_type_; } in compute_type() function in stream_executor::gpu::CudnnRnnDescriptor
Dcuda_blas.cc443 blas::ComputationType compute_type) { in GetScaleType()
2400 cudaDataType_t compute_type = in DoBlasInternalImpl() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4167 auto compute_type = RankedTensorType::get( in matchAndRewrite() local