Searched defs:key_arg (Results 1 – 15 of 15) sorted by relevance
141 const Literal& key_arg = args[0]; in XLA_TEST_F() local172 const Literal& key_arg = args[0]; in XLA_TEST_F() local203 const Literal& key_arg = args[0]; in XLA_TEST_F() local
51 using key_arg = typename KeyArgImpl::template type<K, key_type>; variable
522 using key_arg = typename super_type::template key_arg<K>; variable
541 using key_arg = typename super_type::template key_arg<K>; variable
540 using key_arg = typename super_type::template key_arg<K>; variable
543 using key_arg = typename super_type::template key_arg<K>; variable
482 Tensor key_arg(DT_INT64, TensorShape({})); in StartFlushingGroup() local
1568 Node* key_arg = nullptr; in RewriteHostWhileLoopCond() local1645 Node* key_arg = nullptr; in RewriteHostWhileLoopBody() local