Home
last modified time | relevance | path

Searched refs:xType (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dhlo_utils.cc27 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/
DLinalgToSPIRV.cpp33 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/
Dcheck-expression.cpp301 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/
Dexpression.cpp1233 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/
Dcublas_11_0.inc229 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 …]
Dcublas_10_1.inc229 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 …]
Dcublas_10_2.inc229 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 …]
Dcublas_9_0.inc197 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,
Dcublas_10_0.inc222 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,