Searched refs:call_module (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_call_module_test.py | 62 return xla.call_module([x], module=module, Tout=[x.dtype], Sout=[x.shape])
81 return xla.call_module([x, y],
102 return xla.call_module([x],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 566 def call_module(args, *, module, Tout, Sout, dim_args_spec=()): function
|