Home
last modified time | relevance | path

Searched defs:scale_type (Results 1 – 13 of 13) sorted by relevance

/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()
202 cudaDataType_t BlasLt::MatmulDesc::scale_type() const { in scale_type() function in stream_executor::cuda::BlasLt::MatmulDesc
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dquantize_composite_functions.cc123 TensorType scale_type = RankedTensorType::get({}, rewriter.getF32Type()); in CreateUniformQuantizedTypeParams() local
144 TensorType scale_type = RankedTensorType::get( in CreateUniformQuantizedPerAxisTypeParams() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_util.cc128 se::blas::DataType scale_type = in GetPlanAndAlgorithms() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc82 xla::PrimitiveType scale_type; in CompileImpl() local
/aosp_15_r20/external/libpcap/
H A Dsf-pcap.c150 tstamp_scale_type_t scale_type; member
H A Dsf-pcapng.c204 tstamp_scale_type_t scale_type; /* how to scale */ member
/aosp_15_r20/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/
H A Dstudy.proto394 ScaleType scale_type = 6; field
/aosp_15_r20/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/
H A Dstudy.proto394 ScaleType scale_type = 6; field
/aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1/
H A Dstudy.proto424 ScaleType scale_type = 6; field
/aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1beta1/
H A Dstudy.proto424 ScaleType scale_type = 6; field
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/
H A DGemmHipblaslt.h385 hipDataType scale_type) { in HipBlasLtMatmulDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
H A Dtfr_ops.cc760 auto scale_type = filter_scale_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp272 cudaDataType_t scale_type) { in CuBlasLtMatmulDescriptor()