Searched refs:must_run_on_cpu (Results 1 – 3 of 3) sorted by relevance
27 from tensorflow.python.framework.graph_util_impl import must_run_on_cpu
58 def must_run_on_cpu(node, pin_variables_on_cpu=False): function
16 name: "must_run_on_cpu"