looking for source code of from gen_nn_ops in tensorflow

前端 未结 3 1762
别跟我提以往
别跟我提以往 2020-12-08 13:56

I am new to tensorflow for deep learning and interested in deconvolution (convolution transpose) operation in tensorflow. I need to take a look at the source code for operat

3条回答
  •  清歌不尽
    2020-12-08 14:43

    Unfortunately, TensorFlow code is not easy to read :(

    To make things fast, the python code has to interleave C++ code, which also uses indirect dependencies.

    gen_X functions are generated from their C++ code; to find it, you need to search for Conv2dBackpropInput.

    You can find the registration of the kernel op in ops/nn_ops.cc and concrete implementation in kernels/conv_grad_input_ops.cc.

提交回复
热议问题