Searched refs:input_tokens (Results 1 – 2 of 2) sorted by relevance
139 std::vector<xla::XlaOp> input_tokens; in Compile() local143 input_tokens.push_back(token_or.ValueOrDie()); in Compile()145 xla::XlaOp token = xla::AfterAll(b, input_tokens); in Compile()410 std::vector<xla::XlaOp> input_tokens; in Compile() local414 input_tokens.push_back(token_or.ValueOrDie()); in Compile()416 xla::XlaOp token = xla::AfterAll(b, input_tokens); in Compile()462 std::vector<xla::XlaOp> input_tokens; in Compile() local466 input_tokens.push_back(token_or.ValueOrDie()); in Compile()468 xla::XlaOp token = xla::AfterAll(b, input_tokens); in Compile()
216 std::vector<int> input_tokens(max_sentence_length, pad_token_id); in Preprocess() local220 input_tokens[0] = start_token_id; in Preprocess()230 input_tokens[input_token_index] = token_id; in Preprocess()232 input_tokens[input_token_index] = unknown_token_id; in Preprocess()236 PopulateTensor(input_tokens, input_tensor); in Preprocess()