Searched defs:scale_type (Results 1 – 7 of 7) sorted by relevance
177 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
128 se::blas::DataType scale_type = in GetPlanAndAlgorithms() local
123 TensorType scale_type = RankedTensorType::get({}, rewriter.getF32Type()); in CreateUniformQuantizedTypeParams() local144 TensorType scale_type = RankedTensorType::get( in CreateUniformQuantizedPerAxisTypeParams() local
82 xla::PrimitiveType scale_type; in CompileImpl() local
150 tstamp_scale_type_t scale_type; member
204 tstamp_scale_type_t scale_type; /* how to scale */ member
760 auto scale_type = filter_scale_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local