Cards tested (so far):
Model | Released |
---|---|
Radeon RX 550 | April 2017 |
Radeon RX 560 | April 2017 |
Radeon HD 7800 Series | March 2012 |
def dot_product(x, kernel): | |
""" | |
Wrapper for dot product operation, in order to be compatible with both | |
Theano and Tensorflow | |
Args: | |
x (): input | |
kernel (): weights | |
Returns: | |
""" | |
if K.backend() == 'tensorflow': |
Cards tested (so far):
Model | Released |
---|---|
Radeon RX 550 | April 2017 |
Radeon RX 560 | April 2017 |
Radeon HD 7800 Series | March 2012 |
# Our drawing graph functions. We rely / have borrowed from the following | |
# python libraries: | |
# https://github.com/szagoruyko/pytorchviz/blob/master/torchviz/dot.py | |
# https://github.com/willmcgugan/rich | |
# https://graphviz.readthedocs.io/en/stable/ | |
def draw_graph(start, watch=[]): | |
from graphviz import Digraph |