Searched defs:callback_token (Results 1 – 4 of 4) sorted by relevance
106 tensorflow::Status RunSession(tensorflow::Tensor callback_token, in RunSession()133 std::string RunSessionExpectingSuccess(RandomToken callback_token) { in RunSessionExpectingSuccess()149 RandomToken callback_token = callback_registration.token(); in TEST_F() local157 std::optional<RandomToken> callback_token; in TEST_F() local
36 def serve_slices(callback_token, server_val, max_key, select_fn_initialize_op, argument99 def callback_adapter(callback_token, server_val, *args): argument
133 RandomToken callback_token = RandomToken::FromBytes(callback_token_bytes); in Compute() local
91 absl::string_view select_fn_target_tensor_name) { in enter()