Searched refs:static_argnums_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 138 static_argnums_(std::move(static_argnums)), in PmapFunction()
140 std::sort(static_argnums_.begin(), static_argnums_.end()); in PmapFunction()
180 std::vector<int> static_argnums_; member in jax::PmapFunction
265 if (!ParseArguments(args, kwargs, static_argnums_, arguments).ok()) { in Call()
|
| D | jax_jit.cc | 882 std::vector<int> static_argnums_; member in jax::__anone04173081411::CompiledFunction
923 static_argnums_(std::move(static_argnums)), in CompiledFunction()
927 std::sort(static_argnums_.begin(), static_argnums_.end()); in CompiledFunction()
1146 if (!ParseArguments(args, kwargs, static_argnums_, arguments).ok()) { in Call()
|