Home
last modified time | relevance | path

Searched refs:yType (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dhlo_utils.cc28 TensorType yType = y.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/tensorflow/tensorflow/stream_executor/cuda/
Dcublas_11_0.inc284 cudaDataType yType, int incy,
292 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
299 cudaDataType yType, int incy,
307 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
464 void *y, cudaDataType yType, int incy, cudaDataType executiontype) {
470 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,
522 int incx, void *y, cudaDataType yType,
529 return func_ptr(handle, n, x, xType, incx, y, yType, incy);
615 cudaDataType yType, int incy) {
621 return func_ptr(handle, n, x, xType, incx, y, yType, incy);
[all …]
Dcublas_10_1.inc284 cudaDataType yType, int incy,
292 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
299 cudaDataType yType, int incy,
307 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
464 void *y, cudaDataType yType, int incy, cudaDataType executiontype) {
470 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,
522 int incx, void *y, cudaDataType yType,
529 return func_ptr(handle, n, x, xType, incx, y, yType, incy);
615 cudaDataType yType, int incy) {
621 return func_ptr(handle, n, x, xType, incx, y, yType, incy);
[all …]
Dcublas_10_2.inc284 cudaDataType yType, int incy,
292 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
299 cudaDataType yType, int incy,
307 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
464 void *y, cudaDataType yType, int incy, cudaDataType executiontype) {
470 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,
522 int incx, void *y, cudaDataType yType,
529 return func_ptr(handle, n, x, xType, incx, y, yType, incy);
615 cudaDataType yType, int incy) {
621 return func_ptr(handle, n, x, xType, incx, y, yType, incy);
[all …]
Dcublas_9_0.inc252 cudaDataType yType, int incy,
260 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
267 cudaDataType yType, int incy,
275 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
432 void *y, cudaDataType yType, int incy, cudaDataType executiontype) {
438 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,
Dcublas_10_0.inc277 cudaDataType yType, int incy,
285 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
292 cudaDataType yType, int incy,
300 return func_ptr(handle, n, x, xType, incx, y, yType, incy, result, resultType,
457 void *y, cudaDataType yType, int incy, cudaDataType executiontype) {
463 return func_ptr(handle, n, alpha, alphaType, x, xType, incx, y, yType, incy,