/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
D | hlo_utils.cc | 27 TensorType xType = x.getType().dyn_cast<RankedTensorType>(); in getBroadcastDimensionsAttr() local 29 if (!xType || !yType) return {}; in getBroadcastDimensionsAttr() 30 if (allow_empty && xType == yType) return {}; in getBroadcastDimensionsAttr() 33 auto xRank = xType.getRank(), yRank = yType.getRank(); in getBroadcastDimensionsAttr()
|
/external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/ |
D | LinalgToSPIRV.cpp | 33 Type xType = invocation.getType().cast<ShapedType>().getElementType(); in getLocalInvocationDimSize() local 35 loc, xType, invocation, builder->getI32ArrayAttr({dim})); in getLocalInvocationDimSize()
|
/external/llvm-project/flang/lib/Evaluate/ |
D | check-expression.cpp | 301 auto xType{x.GetType()}; in NonPointerInitializationExpr() local 360 } else if (xType) { in NonPointerInitializationExpr() 363 symbol.name(), xType->AsFortran()); in NonPointerInitializationExpr()
|
/external/llvm-project/flang/lib/Semantics/ |
D | expression.cpp | 1233 DynamicTypeWithLength xType{*dyType}; in Push() local 1235 CHECK(xType.category() == TypeCategory::Character); in Push() 1236 xType.length = in Push() 1248 type_ = std::move(xType); in Push() 1253 static_cast<const DynamicType &>(xType)) { in Push() 1255 if (auto thisLen{ToInt64(xType.LEN())}) { in Push() 1271 type_->length = xType.LEN(); in Push() 1275 type_->length = xType.LEN(); in Push()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cublas_11_0.inc | 229 const void *x, cudaDataType xType, 238 return func_ptr(handle, n, x, xType, incx, result, resultType, executionType); 282 const void *x, cudaDataType xType, 292 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 297 const void *x, cudaDataType xType, 307 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 384 cudaDataType alphaType, void *x, cudaDataType xType, int incx, 391 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, executionType); 463 cudaDataType alphaType, const void *x, cudaDataType xType, int incx, 470 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy, [all …]
|
D | cublas_10_1.inc | 229 const void *x, cudaDataType xType, 238 return func_ptr(handle, n, x, xType, incx, result, resultType, executionType); 282 const void *x, cudaDataType xType, 292 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 297 const void *x, cudaDataType xType, 307 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 384 cudaDataType alphaType, void *x, cudaDataType xType, int incx, 391 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, executionType); 463 cudaDataType alphaType, const void *x, cudaDataType xType, int incx, 470 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy, [all …]
|
D | cublas_10_2.inc | 229 const void *x, cudaDataType xType, 238 return func_ptr(handle, n, x, xType, incx, result, resultType, executionType); 282 const void *x, cudaDataType xType, 292 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 297 const void *x, cudaDataType xType, 307 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 384 cudaDataType alphaType, void *x, cudaDataType xType, int incx, 391 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, executionType); 463 cudaDataType alphaType, const void *x, cudaDataType xType, int incx, 470 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy, [all …]
|
D | cublas_9_0.inc | 197 const void *x, cudaDataType xType, 206 return func_ptr(handle, n, x, xType, incx, result, resultType, executionType); 250 const void *x, cudaDataType xType, 260 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 265 const void *x, cudaDataType xType, 275 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 352 cudaDataType alphaType, void *x, cudaDataType xType, int incx, 359 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, executionType); 431 cudaDataType alphaType, const void *x, cudaDataType xType, int incx, 438 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,
|
D | cublas_10_0.inc | 222 const void *x, cudaDataType xType, 231 return func_ptr(handle, n, x, xType, incx, result, resultType, executionType); 275 const void *x, cudaDataType xType, 285 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 290 const void *x, cudaDataType xType, 300 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType, 377 cudaDataType alphaType, void *x, cudaDataType xType, int incx, 384 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, executionType); 456 cudaDataType alphaType, const void *x, cudaDataType xType, int incx, 463 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,
|