Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcudnn_7_0.inc446 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
451 return func_ptr(convDesc);
455 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t mathType) {
460 return func_ptr(convDesc, mathType);
464 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t *mathType) {
469 return func_ptr(convDesc, mathType);
473 cudnnConvolutionDescriptor_t convDesc, int groupCount) {
478 return func_ptr(convDesc, groupCount);
482 cudnnConvolutionDescriptor_t convDesc, int *groupCount) {
487 return func_ptr(convDesc, groupCount);
[all …]
Dcudnn_6_0.inc437 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
442 return func_ptr(convDesc);
446 cudnnConvolutionDescriptor_t convDesc,
459 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
464 const cudnnConvolutionDescriptor_t convDesc,
477 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
482 const cudnnConvolutionDescriptor_t convDesc,
491 return func_ptr(convDesc, inputTensorDesc, filterDesc, n, c, h, w);
495 cudnnConvolutionDescriptor_t convDesc, int arrayLength, /* nbDims-2 size */
503 return func_ptr(convDesc, arrayLength, padA, filterStrideA, dilationA, mode,
[all …]
Dcudnn_7_3.inc446 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
451 return func_ptr(convDesc);
455 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t mathType) {
460 return func_ptr(convDesc, mathType);
464 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t *mathType) {
469 return func_ptr(convDesc, mathType);
473 cudnnConvolutionDescriptor_t convDesc, int groupCount) {
478 return func_ptr(convDesc, groupCount);
482 cudnnConvolutionDescriptor_t convDesc, int *groupCount) {
487 return func_ptr(convDesc, groupCount);
[all …]
Dcudnn_7_1.inc446 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
451 return func_ptr(convDesc);
455 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t mathType) {
460 return func_ptr(convDesc, mathType);
464 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t *mathType) {
469 return func_ptr(convDesc, mathType);
473 cudnnConvolutionDescriptor_t convDesc, int groupCount) {
478 return func_ptr(convDesc, groupCount);
482 cudnnConvolutionDescriptor_t convDesc, int *groupCount) {
487 return func_ptr(convDesc, groupCount);
[all …]
Dcudnn_7_4.inc446 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
451 return func_ptr(convDesc);
455 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t mathType) {
460 return func_ptr(convDesc, mathType);
464 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t *mathType) {
469 return func_ptr(convDesc, mathType);
473 cudnnConvolutionDescriptor_t convDesc, int groupCount) {
478 return func_ptr(convDesc, groupCount);
482 cudnnConvolutionDescriptor_t convDesc, int *groupCount) {
487 return func_ptr(convDesc, groupCount);
[all …]
Dcudnn_7_6.inc273 const cudnnConvolutionDescriptor_t convDesc,
295 return func_ptr(handle, filterDesc, diffDesc, convDesc, gradDesc,
593 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
598 return func_ptr(convDesc);
602 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t mathType) {
607 return func_ptr(convDesc, mathType);
611 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t *mathType) {
616 return func_ptr(convDesc, mathType);
620 cudnnConvolutionDescriptor_t convDesc, int groupCount) {
625 return func_ptr(convDesc, groupCount);
[all …]
Dcudnn_8_0.inc1025 cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor_t *convDesc) {
1030 return func_ptr(convDesc);
1034 cudnnDestroyConvolutionDescriptor(cudnnConvolutionDescriptor_t convDesc) {
1039 return func_ptr(convDesc);
1043 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t mathType) {
1048 return func_ptr(convDesc, mathType);
1052 cudnnConvolutionDescriptor_t convDesc, cudnnMathType_t *mathType) {
1057 return func_ptr(convDesc, mathType);
1061 cudnnConvolutionDescriptor_t convDesc, int groupCount) {
1066 return func_ptr(convDesc, groupCount);
[all …]
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1082 …AddConvLayerWithDepthwiseConv(const onnx::NodeProto& node, const Convolution2dDescriptor& convDesc) in AddConvLayerWithDepthwiseConv() argument
1087 desc.m_PadLeft = convDesc.m_PadLeft; in AddConvLayerWithDepthwiseConv()
1088 desc.m_PadRight = convDesc.m_PadRight; in AddConvLayerWithDepthwiseConv()
1089 desc.m_PadTop = convDesc.m_PadTop; in AddConvLayerWithDepthwiseConv()
1090 desc.m_PadBottom = convDesc.m_PadBottom; in AddConvLayerWithDepthwiseConv()
1091 desc.m_StrideX = convDesc.m_StrideX; in AddConvLayerWithDepthwiseConv()
1092 desc.m_StrideY = convDesc.m_StrideY; in AddConvLayerWithDepthwiseConv()
1093 desc.m_BiasEnabled = convDesc.m_BiasEnabled; in AddConvLayerWithDepthwiseConv()
DOnnxParser.hpp119 …ayerWithDepthwiseConv(const onnx::NodeProto& node, const armnn::Convolution2dDescriptor& convDesc);