Searched refs:dimA (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 123 const int dimA[], const int strideA[]) {
128 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
133 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
139 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
144 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
150 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
1352 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
1359 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
|
| D | cudnn_8_0.inc | 150 const int dimA[], const int strideA[]) {
155 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
160 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
166 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
171 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
177 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
853 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
860 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
1986 const int dimA[], const cudnnSeqDataAxis_t axes[],
1993 return func_ptr(seqDataDesc, dataType, nbDims, dimA, axes, seqLengthArraySize,
[all …]
|
| D | cudnn_7_0.inc | 134 const int dimA[], const int strideA[]) {
139 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
144 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
150 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
155 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
161 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
1473 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
1480 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
|
| D | cudnn_7_6.inc | 134 const int dimA[], const int strideA[]) {
139 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
144 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
150 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
155 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
161 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
1779 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
1786 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
2668 const int dimA[], const cudnnSeqDataAxis_t axes[],
2675 return func_ptr(seqDataDesc, dataType, nbDims, dimA, axes, seqLengthArraySize,
[all …]
|
| D | cudnn_7_3.inc | 134 const int dimA[], const int strideA[]) {
139 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
144 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
150 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
155 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
161 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
1473 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
1480 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
|
| D | cudnn_7_1.inc | 134 const int dimA[], const int strideA[]) {
139 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
144 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
150 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
155 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
161 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
1473 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
1480 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
|
| D | cudnn_7_4.inc | 134 const int dimA[], const int strideA[]) {
139 return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA);
144 cudnnDataType_t dataType, int nbDims, const int dimA[]) {
150 return func_ptr(tensorDesc, format, dataType, nbDims, dimA);
155 cudnnDataType_t *dataType, int *nbDims, int dimA[], int strideA[]) {
161 return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA);
1614 cudnnDataType_t dataType, const int nbDims, const int dimA[]) {
1621 return func_ptr(stDesc, samplerType, dataType, nbDims, dimA);
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 2549 int64_t dimA = a[i]; in isValidShapeCast() local
2551 while (dimB < dimA && j < rankB) in isValidShapeCast()
2553 if (dimA != dimB) in isValidShapeCast()
|