J'essaie de trouver où se trouve la mise en œuvre d'une opération Conv2D réelle afin de pouvoir évaluer les modèles d'accès à la mémoire. En traçant les choses, il semble que l'exécution d'une opération Conv2D entre dans Eigen avec un appel de fonction contract(). Le problème est que je n'arrive pas à trouver la définition ou la déclaration de la fonction dans la source TensorFlow ou Eigen.

Quelles fonctions sont en grande partie responsables de l'exécution d'une opération Conv2D dans TensorFlow? J'aimerais voir comment il est paralysé, quel est le modèle général d'accès à la mémoire et comment les calculs bruts sont effectués.

Cette requête concerne spécifiquement le processeur, car j'ai déjà étudié dans une certaine mesure l'exécution du GPU.

1
Armando Montanez 4 août 2017 à 22:45

2 réponses

Meilleure réponse

Après quelques recherches, j'ai trouvé que la mise en œuvre réelle du CPU Conv2D se trouve dans deep_conv2d.cc.

1
Armando Montanez 18 août 2017 à 02:38

Je pense que Conv2dCPU est implémenté dans ce fichier en utilisant Eigen conv ops Line 61 et suivantes

0
Ishant Mrinal 4 août 2017 à 21:25