Searched refs:pin_variables_on_cpu (Results 1 – 2 of 2) sorted by relevance
17 …argspec: "args=[\'node\', \'pin_variables_on_cpu\'], varargs=None, keywords=None, defaults=[\'Fals…
72 def must_run_on_cpu(node, pin_variables_on_cpu=False): argument92 if pin_variables_on_cpu and _is_variable_op(node_def.op):