Searched refs:GetTensorListPushIndex (Results 1 – 4 of 4) sorted by relevance
76 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &push_index)); in Compile()
53 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index);
373 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(list, &push_index)); in Compile()
160 Status GetTensorListPushIndex(xla::XlaOp list, xla::XlaOp* push_index) { in GetTensorListPushIndex() function