Searched defs:input_ids (Results 1 – 13 of 13) sorted by relevance
35 std::vector<int> input_ids; member
99 std::vector<int> input_ids(kMaxSeqLen, 0); in Preprocess() local
126 repeated int64 input_ids = 2; field
142 const std::vector<int>& input_ids) { in MaybeLockVariableInputMutexesInOrder()
259 std::vector<int> input_ids(tokens.size()); in Preprocess() local
110 const flatbuffers::Vector<int32_t> *input_ids() const { in input_ids() function360 const flatbuffers::Vector<int32_t> *input_ids() const { in input_ids() function
2306 const std::vector<tensorflow::int64>& input_ids,2326 const std::vector<tensorflow::int64>& input_ids,2527 const std::vector<tensorflow::int64>& input_ids,2566 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local2620 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local2671 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(input_tensors); local3160 std::vector<tensorflow::int64> input_ids = MakeTensorIDList(inputs); local
242 std::vector<int> input_ids(inputs, inputs + len); in TF_MaybeLockVariableInputMutexesInOrder() local
422 std::vector<int64_t> input_ids; in ExecuteProgram() local
229 std::vector<int64> input_ids(inputs.size()); in RecordOperation() local
2802 const std::vector<int>& input_ids, const std::vector<int>& output_ids, in BuildModelEnforceIO()2893 std::vector<int> input_ids; member
4590 std::vector<int64> input_ids) { in FilterInputsForShmemTranspose()
2574 repeated int64 input_ids = 2; field