Searched defs:push_index (Results 1 – 3 of 3) sorted by relevance
132 Status BuildNonNestedTensorList(xla::XlaOp buffer, xla::XlaOp push_index, in BuildNonNestedTensorList()160 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index) { in GetTensorListPushIndex()172 Status SetTensorListPushIndex(xla::XlaOp list, xla::XlaOp push_index, in SetTensorListPushIndex()364 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPushBack() local429 xla::XlaOp push_index = xla::GetTupleElement(list, list_tuple_size - 1); in ExecuteTensorListPopBack() local540 Status ExecuteTensorListFromTensor(int push_index, xla::XlaOp tensor, in ExecuteTensorListFromTensor()
75 xla::XlaOp push_index; in Compile() local
372 xla::XlaOp push_index; in Compile() local