Home
last modified time | relevance | path

Searched refs:dynamic_positional_args_treedef (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc76 return std::tie(dynamic_positional_args_treedef, keyword_args, in operator ==()
78 std::tie(other.dynamic_positional_args_treedef, other.keyword_args, in operator ==()
133 signature_str.reserve(dynamic_positional_args_treedef.size()); in DebugString()
134 for (auto& tree_def : dynamic_positional_args_treedef) { in DebugString()
148 dynamic_positional_args_treedef.size(), " positional args.\n", in DebugString()
158 s.dynamic_positional_args_treedef.data(), in AbslHashValue()
159 s.dynamic_positional_args_treedef.size()); in AbslHashValue()
193 arguments.signature.dynamic_positional_args_treedef.reserve( in ParseArguments()
202 arguments.signature.dynamic_positional_args_treedef.push_back(pytree_def); in ParseArguments()
Djax_jit.h82 std::vector<xla::PyTreeDef> dynamic_positional_args_treedef; member