Searched refs:num_resource_args (Results 1 – 4 of 4) sorted by relevance
181 int num_resource_args = 0; in Compile() local197 num_resource_args++; in Compile()322 xla::GetTupleElement(outputs, output_types_.size() + num_resource_args); in Compile()
91 int num_resource_args = 0; in Compile() local107 num_resource_args++; in Compile()337 xla::GetTupleElement(outputs, output_types_.size() + num_resource_args); in Compile()
90 int num_resource_args, Node** result) { in MakeXlaCompiledKernel() argument96 AddNodeAttr(kXlaNumResourceArgsAttr, num_resource_args, &call_node); in MakeXlaCompiledKernel()
2111 int num_resource_args = 0; in Verify() local2114 ++num_resource_args; in Verify()2118 if (indices.size() != num_resource_args) in Verify()2123 << num_resource_args << "), but got " << indices.size(); in Verify()