Where is the code Tensorflow uses to implement RMSProp

user1245262 :

I've cloned Tensorflow from https://github.com/tensorflow/tensorflow.git and am trying to find the code where RMS_Prop is implemented.

I have found the file tensorflow/tensorflow/python/training/rmsprop.py, which has calls to training_ops.apply_centered_rms_prop and to training_ops.apply_rms_prop, which seem to be the methods I'm looking for.

I can see that rmsprop.py imports training ops with this import statement:

from tensorflow.python.training import training_ops

but, when I look at training_ops.py, all I see is the following code:

"""Python wrappers for training ops."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow.python.training import gen_training_ops  # pylint: disable=unused-import
# go/tf-wildcard-import
# pylint: disable=wildcard-import
from tensorflow.python.training.gen_training_ops import *
# pylint: enable=wildcard-import

I do not see gen_training_ops in the tensorflow/tensorflow/python/training directory and the __init__ file is empty.

What is it I'm not understanding here, and where can I find the actual code that directly implements rmsprop?

RadioSilence :

The main computational code is implemented in C++; the Python layer abstracts this. The files you are interested in are the CPU kernel and GPU kernel. The kernel for RMSProp starts on lines 500 and 408 respectively.

Guess you like

Origin http://10.200.1.11:23101/article/api/json?id=383503&siteId=1

Where is the code Tensorflow uses to implement RMSProp

[Natural Language NLP] TensorFlow uses LSTM to implement text matching tasks

[Natural Language NLP] TensorFlow uses LSTM to implement sentiment analysis tasks

Android uses WindowManager to implement floating window and source code analysis

Android uses WindowManager to implement floating window and source code analysis

TensorFlow used to implement a simple identification code verification process

Linux uses pyinstaller to package tensorflow's python code into executable files

tensorflow implement XOR

Android uses the framework to implement two lines of code to change the avatar from a local picture or taking a photo?

Lituo884. Uncommon words in two sentences--Python uses 2 lines of code to implement

The father of Redis uses pure C language code to implement the Telegram Bot framework

python deconvolution (DeConv) tensorflow deconvolution (DeConv) (implementation principle + handwriting) Tensorflow deconvolution (DeConv) implementation principle + handwritten python code to implement deconvolution (DeConv)

Python uses one hundred lines of code to implement a scientific calculator (with complete code link) - the first course assignment for software engineering

Где код Tensorflow использует для реализации RMSProp

Java uses Http to implement post

Android uses LruCache to implement LruCacheSet

Angularjs uses directive to implement tree

Angularjs uses directive to implement tree

React uses hook to implement TodoList

tensorflow (2) Using tensorflow to implement linear regression

Implement Imagenet training with Tensorflow 2.0

MySQL uses IF function to dynamically execute where conditions

Summary of scenarios where java uses multithreading

Golang uses chan (select) and goroutine to implement: the execution of time-consuming code blocks does not affect the timely response of http services

Where is TensorFlow going? PyTorch accounts for 80% of academia

Tensorflow saves and uses the training model

Manually implement the verification code

Code to implement TextCNN classification

PyODPS DataFrame code where to run

Solving problems where the code (python)