Searched defs:input_ids (Results 1 – 9 of 9) sorted by relevance
35 std::vector<int> input_ids; member
99 std::vector<int> input_ids(kMaxSeqLen, 0); in Preprocess() local
143 const std::vector<int>& input_ids) { in MaybeLockVariableInputMutexesInOrder()
105 const flatbuffers::Vector<int32_t> *input_ids() const { in input_ids() function287 const flatbuffers::Vector<int32_t> *input_ids() const { in input_ids() function
259 std::vector<int> input_ids(tokens.size()); in Preprocess() local
2294 const std::vector<tensorflow::int64>& input_ids,2314 const std::vector<tensorflow::int64>& input_ids,2515 const std::vector<tensorflow::int64>& input_ids,2554 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local2608 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local2659 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local3148 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(inputs); local
422 std::vector<int64_t> input_ids; in ExecuteProgram() local
229 std::vector<int64> input_ids(inputs.size()); in RecordOperation() local
5073 std::vector<int64> input_ids) { in FilterInputsForShmemTranspose()